diff options
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/ensure_bias_vectors.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/ensure_bias_vectors.cc | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/ensure_bias_vectors.cc b/tensorflow/contrib/lite/toco/graph_transformations/ensure_bias_vectors.cc index e80ed036b3..72b1dda3be 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/ensure_bias_vectors.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/ensure_bias_vectors.cc @@ -62,17 +62,20 @@ bool ProcessLinearOperator(Model* model, Operator* op) { } } // namespace -bool EnsureBiasVectors::Run(Model* model, std::size_t op_index) { +::tensorflow::Status EnsureBiasVectors::Run(Model* model, std::size_t op_index, + bool* modified) { + *modified = false; auto* op = model->operators[op_index].get(); if (op->type == OperatorType::kConv || op->type == OperatorType::kDepthwiseConv || op->type == OperatorType::kFullyConnected) { if (ProcessLinearOperator(model, op)) { AddMessageF("Added bias vector to %s as %s", LogName(*op), op->inputs[2]); - return true; + *modified = true; + return ::tensorflow::Status::OK(); } } - return false; + return ::tensorflow::Status::OK(); } } // namespace toco |