diff options
author | Yu-Cheng Ling <ycling@google.com> | 2018-10-09 11:38:15 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 11:48:46 -0700 |
commit | 12e164d1e7c0b197f06d5d3c2ed26318b89b5e4c (patch) | |
tree | d2f0b6ba463baff8e3607575f41d3655762f3d14 /tensorflow/contrib/lite/toco/graph_transformations/convert_trivial_addn_to_add.cc | |
parent | 931353c5f79c2d419afb3a5ecac59184c5558351 (diff) |
Return ::tensorflow::Status in Toco Graph Transformations.
PiperOrigin-RevId: 216392908
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/convert_trivial_addn_to_add.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/convert_trivial_addn_to_add.cc | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/convert_trivial_addn_to_add.cc b/tensorflow/contrib/lite/toco/graph_transformations/convert_trivial_addn_to_add.cc index dcaaddbf3b..d7cacf77f4 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/convert_trivial_addn_to_add.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/convert_trivial_addn_to_add.cc @@ -20,10 +20,13 @@ namespace toco { // This pass will convert an AddN operator with only 2 inputs into a regular Add // operator, to which more optimizations may apply. -bool ConvertTrivialAddNToAdd::Run(Model* model, std::size_t op_index) { +::tensorflow::Status ConvertTrivialAddNToAdd::Run(Model* model, + std::size_t op_index, + bool* modified) { + *modified = false; auto addn_it = model->operators.begin() + op_index; if (addn_it->get()->type != OperatorType::kAddN) { - return false; + return ::tensorflow::Status::OK(); } AddNOperator* addn_op = static_cast<AddNOperator*>(addn_it->get()); CHECK_GE(addn_op->inputs.size(), 2); @@ -31,7 +34,7 @@ bool ConvertTrivialAddNToAdd::Run(Model* model, std::size_t op_index) { // We only reduce AddN with N=2 to a regular Add. if (addn_op->inputs.size() != 2) { - return false; + return ::tensorflow::Status::OK(); } // Copy inputs & outputs to regular Add. @@ -45,7 +48,8 @@ bool ConvertTrivialAddNToAdd::Run(Model* model, std::size_t op_index) { addn_it = add_it + 1; CHECK_EQ(addn_it->get(), addn_op); model->operators.erase(addn_it); - return true; + *modified = true; + return ::tensorflow::Status::OK(); } } // namespace toco |