diff options
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/propagate_default_min_max.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/propagate_default_min_max.cc | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/propagate_default_min_max.cc b/tensorflow/contrib/lite/toco/graph_transformations/propagate_default_min_max.cc index cd078ef189..3cf191436d 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/propagate_default_min_max.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/propagate_default_min_max.cc @@ -39,7 +39,10 @@ bool SupportsMinMax(const Array& array) { // When provided a set of min/max values for uint8 arrays this will rescale // the values for other data types as required and preserving the floating point // range within the new type. -bool PropagateDefaultMinMax::Run(Model* model, std::size_t op_index) { +::tensorflow::Status PropagateDefaultMinMax::Run(Model* model, + std::size_t op_index, + bool* modified) { + *modified = false; const auto it = model->operators.begin() + op_index; const auto* op = it->get(); @@ -61,7 +64,8 @@ bool PropagateDefaultMinMax::Run(Model* model, std::size_t op_index) { } } - return did_change; + *modified = did_change; + return ::tensorflow::Status::OK(); } // Sets the min/max on the given array, adjusting the reference_minmax for the |