Merge pull request #12805 from tensor-tang/fix/op/elewise_add

fix SEGV element wise add at debug mode
dataset_flowers_-_change_md5
Tao Luo 7 years ago committed by GitHub
commit 8f9f414a14
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -80,6 +80,9 @@ inline framework::DDim trim_trailing_singular_dims(
for (int i = 0; i < actual_dims_size; ++i) {
trim_dims[i] = dims[i];
}
if (trim_dims.size() == 0) {
return framework::DDim(framework::make_dim());
}
framework::DDim actual_dims = framework::make_ddim(trim_dims);
return actual_dims;
}

Loading…
Cancel
Save