experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use TensorShape::num_elements() instead of loop

Browse Source
This commit is contained in:
Ben Barsdell 2019-04-24 10:46:47 -07:00
parent 7008e41f18
commit a9261f1243
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/core/grappler/optimizers/constant_folding.cc
View File

@ -2394,15 +2394,9 @@ bool ConstantFolding::IsReductionWithConstantIndices(
!reductions_indices->attr().count("value")) { !reductions_indices->attr().count("value")) {
return false; return false;
} }
const TensorProto& reduction_indices_tensor = const TensorShapeProto& reduction_indices_shape =
reductions_indices->attr().at("value").tensor(); reductions_indices->attr().at("value").tensor().tensor_shape();
*indices_is_empty = false; *indices_is_empty = TensorShape(reduction_indices_shape).num_elements() == 0;
for (const auto& dim : reduction_indices_tensor.tensor_shape().dim()) {
if (dim.size() == 0) {
*indices_is_empty = true;
break;
}
}
return true; return true;
} }