From 12e164d1e7c0b197f06d5d3c2ed26318b89b5e4c Mon Sep 17 00:00:00 2001 From: Yu-Cheng Ling Date: Tue, 9 Oct 2018 11:38:15 -0700 Subject: Return ::tensorflow::Status in Toco Graph Transformations. PiperOrigin-RevId: 216392908 --- .../lite/toco/graph_transformations/drop_im2col_arrays.cc | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'tensorflow/contrib/lite/toco/graph_transformations/drop_im2col_arrays.cc') diff --git a/tensorflow/contrib/lite/toco/graph_transformations/drop_im2col_arrays.cc b/tensorflow/contrib/lite/toco/graph_transformations/drop_im2col_arrays.cc index f7fd878b7e..7d66ea5dd2 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/drop_im2col_arrays.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/drop_im2col_arrays.cc @@ -19,15 +19,17 @@ limitations under the License. namespace toco { -bool DropIm2colArrays::Run(Model* model, std::size_t op_index) { +::tensorflow::Status DropIm2colArrays::Run(Model* model, std::size_t op_index, + bool* modified) { + *modified = false; auto conv_it = model->operators.begin() + op_index; if (conv_it->get()->type != OperatorType::kConv) { - return false; + return ::tensorflow::Status::OK(); } auto* conv_op = static_cast(conv_it->get()); if (conv_op->outputs.size() < 2) { // Conv op does not have im2col. - return false; + return ::tensorflow::Status::OK(); } // Drop the im2col array. @@ -36,7 +38,8 @@ bool DropIm2colArrays::Run(Model* model, std::size_t op_index) { conv_op->outputs.resize(1); AddMessageF("Dropped an im2col array for %s", LogName(*conv_op)); - return true; + *modified = true; + return ::tensorflow::Status::OK(); } } // namespace toco -- cgit v1.2.3