diff options
| author |  Yu-Cheng Ling <ycling@google.com> | 2018-09-27 16:19:09 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 16:33:31 -0700 |
| commit | 2330933ddd0b29ad206e351c9120e621cdaf6312 (patch) | |
| tree | c617ed415b787d276619fcd19f32b6aa1e133453 /tensorflow/contrib/lite/python/lite_test.py | |
| parent | 5f67bf69d3f53d1cd3bb86ebeeb03ea2bba5911b (diff) | |
Rename TFLite Extended -> TFLite Flex
PiperOrigin-RevId: 214854303
Diffstat (limited to 'tensorflow/contrib/lite/python/lite_test.py')
| -rw-r--r-- | tensorflow/contrib/lite/python/lite_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/python/lite_test.py b/tensorflow/contrib/lite/python/lite_test.py index 7b0df01d1d..d243a494f6 100644 --- a/tensorflow/contrib/lite/python/lite_test.py +++ b/tensorflow/contrib/lite/python/lite_test.py @@ -412,7 +412,7 @@ class FromSessionTest(test_util.TensorFlowTestCase): # Ensure that the quantized weights tflite model is smaller. self.assertTrue(len(quantized_tflite) < len(float_tflite)) - def testExtendedMode(self): + def testFlexMode(self): in_tensor = array_ops.placeholder( shape=[1, 16, 16, 3], dtype=dtypes.float32) out_tensor = in_tensor + in_tensor @@ -421,7 +421,7 @@ class FromSessionTest(test_util.TensorFlowTestCase): # Convert model and ensure model is not None. converter = lite.TFLiteConverter.from_session(sess, [in_tensor], [out_tensor]) - converter.converter_mode = lite.ConverterMode.TOCO_EXTENDED_ALL + converter.converter_mode = lite.ConverterMode.TOCO_FLEX_ALL tflite_model = converter.convert() self.assertTrue(tflite_model) |