diff options
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc b/tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc index 6961e23690..8b0bc2d865 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc @@ -142,7 +142,7 @@ bool GraphTransformationsPass(int increment, Model* model, for (const auto& transformation : transformations) { CHECK(!changed_now); CHECK(transformation->Messages().empty()); - changed_now = transformation->Run(model, op_index); + CHECK(transformation->Run(model, op_index, &changed_now).ok()); const char* made_a_change_msg = changed_now ? "made a change" : "did NOT make a change"; const int log_level = |