diff options
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/lstm_utils.h')
| -rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/lstm_utils.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/lstm_utils.h b/tensorflow/contrib/lite/toco/graph_transformations/lstm_utils.h index 4a9974ed4e..1c32a78169 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/lstm_utils.h +++ b/tensorflow/contrib/lite/toco/graph_transformations/lstm_utils.h @@ -51,10 +51,10 @@ enum ExtendedLstmCellInputs { }; enum ExtendedLstmCellOutputs { - kScratchBufferTensor = 0, - kOutputStateTensor = 1, - kCellStateTensor = 2, - kOutputTensor = 3 + kOutputStateTensor = 0, + kCellStateTensor = 1, + kOutputTensor = 2, + kExtendedLstmOutputCount = 3 }; // Create optional array used for optional tensor in ExtendedLstmCell inputs. |