diff options
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/hardcode_min_max.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/hardcode_min_max.cc | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/hardcode_min_max.cc b/tensorflow/contrib/lite/toco/graph_transformations/hardcode_min_max.cc index 3114fa93e8..72df53548b 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/hardcode_min_max.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/hardcode_min_max.cc @@ -372,7 +372,9 @@ bool HardcodeMinMaxForLstmCell(Model* model, Operator* op) { } } // namespace -bool HardcodeMinMax::Run(Model* model, std::size_t op_index) { +::tensorflow::Status HardcodeMinMax::Run(Model* model, std::size_t op_index, + bool* modified) { + *modified = false; auto it = model->operators.begin() + op_index; auto* op = it->get(); bool changed = false; @@ -467,7 +469,8 @@ bool HardcodeMinMax::Run(Model* model, std::size_t op_index) { if (changed) { AddMessageF("Hardcoded min-max through %s", LogName(*op)); } - return changed; + *modified = changed; + return ::tensorflow::Status::OK(); } } // namespace toco |