diff options
author | 2018-09-07 15:59:25 -0700 | |
---|---|---|
committer | 2018-09-07 16:03:30 -0700 | |
commit | 8bd0920af5f9bea05470f9a73f3c688962cd8cb5 (patch) | |
tree | e158f950e14b303a5e50799d7c61d0e6bdc9c827 /tensorflow/contrib/lite/tools | |
parent | 0b3786c1da9ea9532c48a11974886007794202f2 (diff) |
Bidi lstm supports hybrid eval.
PiperOrigin-RevId: 212050890
Diffstat (limited to 'tensorflow/contrib/lite/tools')
-rw-r--r-- | tensorflow/contrib/lite/tools/optimize/quantize_weights.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/tools/optimize/quantize_weights.cc b/tensorflow/contrib/lite/tools/optimize/quantize_weights.cc index 692efb9029..b863108aa4 100644 --- a/tensorflow/contrib/lite/tools/optimize/quantize_weights.cc +++ b/tensorflow/contrib/lite/tools/optimize/quantize_weights.cc @@ -141,6 +141,7 @@ bool IsHybridEvaluationOp(const OperatorT* op, const BuiltinOperator& op_code) { op_code == BuiltinOperator_CONV_2D || op_code == BuiltinOperator_SVDF || op_code == BuiltinOperator_EMBEDDING_LOOKUP || op_code == BuiltinOperator_RNN || + op_code == BuiltinOperator_BIDIRECTIONAL_SEQUENCE_LSTM || op_code == BuiltinOperator_BIDIRECTIONAL_SEQUENCE_RNN || op_code == BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_LSTM || op_code == BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_RNN) { |