aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/toco/graph_transformations/remove_trivial_concatenation.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/remove_trivial_concatenation.cc')
-rw-r--r--tensorflow/contrib/lite/toco/graph_transformations/remove_trivial_concatenation.cc12
1 files changed, 8 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/remove_trivial_concatenation.cc b/tensorflow/contrib/lite/toco/graph_transformations/remove_trivial_concatenation.cc
index 3ceb93d8ee..faaa2a828e 100644
--- a/tensorflow/contrib/lite/toco/graph_transformations/remove_trivial_concatenation.cc
+++ b/tensorflow/contrib/lite/toco/graph_transformations/remove_trivial_concatenation.cc
@@ -25,16 +25,20 @@ limitations under the License.
namespace toco {
-bool RemoveTrivialConcatenation::Run(Model* model, std::size_t op_index) {
+::tensorflow::Status RemoveTrivialConcatenation::Run(Model* model,
+ std::size_t op_index,
+ bool* modified) {
+ *modified = false;
const auto concat_it = model->operators.begin() + op_index;
auto* concat_op = concat_it->get();
if (concat_op->type != OperatorType::kConcatenation) {
- return false;
+ return ::tensorflow::Status::OK();
}
if (concat_op->inputs.size() != 1) {
- return false;
+ return ::tensorflow::Status::OK();
}
- return RemoveTrivialPassthroughOp(this, model, op_index);
+ *modified = RemoveTrivialPassthroughOp(this, model, op_index);
+ return ::tensorflow::Status::OK();
}
} // namespace toco