commit | 7d942e00ec5bfc01fd55825d8dded2e02f965c18 | [log] [tgz] |
---|---|---|
author | Karim Nosir <karimnosseir@google.com> | Wed Jan 01 19:31:38 2020 -0800 |
committer | TensorFlower Gardener <gardener@tensorflow.org> | Wed Jan 01 19:34:33 2020 -0800 |
tree | 9253284a3b8337986bc5dda2c5d8232dd3b8fcb5 | |
parent | 2712ab1563c140416ac6daa8b82b2a7b4ace119c [diff] |
Set the option to use experimental converter before the convert line. PiperOrigin-RevId: 287781634 Change-Id: I558c5cf0debfb85b3daa96de65466c1e0f764d3d
diff --git a/tensorflow/lite/experimental/examples/lstm/unidirectional_sequence_lstm_test.py b/tensorflow/lite/experimental/examples/lstm/unidirectional_sequence_lstm_test.py index f27086a..48e434a 100644 --- a/tensorflow/lite/experimental/examples/lstm/unidirectional_sequence_lstm_test.py +++ b/tensorflow/lite/experimental/examples/lstm/unidirectional_sequence_lstm_test.py
@@ -221,8 +221,8 @@ """ converter = tf.lite.TFLiteConverter.from_session(sess, [input_tensor], [output_tensor]) - tflite = converter.convert() converter.experimental_new_converter = use_mlir_converter + tflite = converter.convert() interpreter = tf.lite.Interpreter(model_content=tflite)
diff --git a/tensorflow/lite/experimental/examples/lstm/unidirectional_sequence_rnn_test.py b/tensorflow/lite/experimental/examples/lstm/unidirectional_sequence_rnn_test.py index bb16191..47799b7 100644 --- a/tensorflow/lite/experimental/examples/lstm/unidirectional_sequence_rnn_test.py +++ b/tensorflow/lite/experimental/examples/lstm/unidirectional_sequence_rnn_test.py
@@ -216,8 +216,8 @@ """ converter = tf.lite.TFLiteConverter.from_session(sess, [input_tensor], [output_tensor]) - tflite = converter.convert() converter.experimental_new_converter = use_mlir_converter + tflite = converter.convert() interpreter = tf.lite.Interpreter(model_content=tflite) interpreter.allocate_tensors()