diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-06 21:00:57 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-06 21:06:15 -0700 |
commit | 7fa6a6b42bc9d562e2b1cc765ca78d281b51f734 (patch) | |
tree | 19398616e50223b0876e6e9a987cc9908d404592 /tensorflow/contrib/lite/kernels/unidirectional_sequence_lstm_test.cc | |
parent | e93a18954689b6d522560f5273f6d3320d545b2e (diff) |
Add SequenceLSTMOptions to schema to decouple the sequential Op from the LSTM.
PiperOrigin-RevId: 216066634
Diffstat (limited to 'tensorflow/contrib/lite/kernels/unidirectional_sequence_lstm_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/unidirectional_sequence_lstm_test.cc | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/tensorflow/contrib/lite/kernels/unidirectional_sequence_lstm_test.cc b/tensorflow/contrib/lite/kernels/unidirectional_sequence_lstm_test.cc index cd3aac0532..c97b0fdd61 100644 --- a/tensorflow/contrib/lite/kernels/unidirectional_sequence_lstm_test.cc +++ b/tensorflow/contrib/lite/kernels/unidirectional_sequence_lstm_test.cc @@ -110,11 +110,12 @@ class UnidirectionalLSTMOpModel : public SingleOpModel { output_ = AddOutput(TensorType_FLOAT32); - SetBuiltinOp(BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_LSTM, - BuiltinOptions_LSTMOptions, - CreateLSTMOptions(builder_, ActivationFunctionType_TANH, - cell_clip, proj_clip) - .Union()); + SetBuiltinOp( + BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_LSTM, + BuiltinOptions_UnidirectionalSequenceLSTMOptions, + CreateUnidirectionalSequenceLSTMOptions( + builder_, ActivationFunctionType_TANH, cell_clip, proj_clip) + .Union()); BuildInterpreter(input_shapes); } |