Fix sum op fails as no memory in tensor(#20602)

test=develop

Signed-off-by: zhaoyuchen <zhaoyuchen01@baidu.com>
revert-20712-fix_depthwise_conv
zhaoyuchen2018 5 years ago committed by GitHub
parent 48b27229a8
commit 8314e64a8b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -163,8 +163,10 @@ void SumToLoDTensor(const framework::ExecutionContext &context) {
for (int i = start; i < in_num; ++i) {
if (in_vars[i]->IsType<framework::LoDTensor>()) {
auto &in_i = in_vars[i]->Get<framework::LoDTensor>();
in_data.emplace_back(in_i.data<T>());
lod_length = in_i.numel();
if (lod_length && in_i.IsInitialized()) {
in_data.emplace_back(in_i.data<T>());
}
} else if (in_vars[i]->IsType<framework::SelectedRows>()) {
selectrow_index.push_back(i);
}

Loading…
Cancel
Save