diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-12 16:26:13 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-12 16:28:23 -0700 |
| commit | cc108a73af35b407bf9bf51e679e5884b309964b (patch) | |
| tree | 86a60734533509dedffff49e3c1819ab8dc37d10 | |
| parent | 0195d6b4fbbe948914d0045d19eec8fcef1211f5 (diff) | |
Add support for RNN state array of type tf.identity.
PiperOrigin-RevId: 192689747
| -rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/remove_unused_op.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/remove_unused_op.cc b/tensorflow/contrib/lite/toco/graph_transformations/remove_unused_op.cc index aa2c293382..8e6aaf544a 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/remove_unused_op.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/remove_unused_op.cc @@ -47,7 +47,8 @@ bool RemoveUnusedOp::Run(Model* model, std::size_t op_index) { bool found_output_as_rnn_state_array = false; for (const auto& rnn_state : model->flags.rnn_states()) { if (output == rnn_state.state_array()) { - CHECK(op->type == OperatorType::kFill); + CHECK(op->type == OperatorType::kFill || + op->type == OperatorType::kTensorFlowIdentity); found_output_as_rnn_state_array = true; break; } |