diff options
author | Yu-Cheng Ling <ycling@google.com> | 2018-09-27 14:25:18 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 14:30:11 -0700 |
commit | d0397c3314600da0c9cdc300ae87483331d54298 (patch) | |
tree | 8ec45f483facaa2219324a19a7f74e38e2e0a5e2 /tensorflow/contrib/lite/python | |
parent | cc83067469bc30bba55932c587f31ef68f15792f (diff) |
Rename TFLite Eager delegate -> Flex delegate
PiperOrigin-RevId: 214835588
Diffstat (limited to 'tensorflow/contrib/lite/python')
-rw-r--r-- | tensorflow/contrib/lite/python/convert.py | 6 | ||||
-rw-r--r-- | tensorflow/contrib/lite/python/lite_test.py | 2 |
2 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/python/convert.py b/tensorflow/contrib/lite/python/convert.py index 627be8f44f..73a420c47b 100644 --- a/tensorflow/contrib/lite/python/convert.py +++ b/tensorflow/contrib/lite/python/convert.py @@ -241,10 +241,10 @@ def build_toco_convert_protos(input_tensors, toco.dump_graphviz_dir = dump_graphviz_dir toco.dump_graphviz_include_video = dump_graphviz_video if converter_mode == ConverterMode.TOCO_EXTENDED: - toco.allow_eager_ops = True + toco.allow_flex_ops = True elif converter_mode == ConverterMode.TOCO_EXTENDED_ALL: - toco.allow_eager_ops = True - toco.force_eager_ops = True + toco.allow_flex_ops = True + toco.force_flex_ops = True model = _model_flags_pb2.ModelFlags() model.change_concat_input_ranges = change_concat_input_ranges diff --git a/tensorflow/contrib/lite/python/lite_test.py b/tensorflow/contrib/lite/python/lite_test.py index 33f8fc1e8c..7b0df01d1d 100644 --- a/tensorflow/contrib/lite/python/lite_test.py +++ b/tensorflow/contrib/lite/python/lite_test.py @@ -432,7 +432,7 @@ class FromSessionTest(test_util.TensorFlowTestCase): interpreter.allocate_tensors() self.assertIn( 'Regular TensorFlow ops are not supported by this interpreter. Make ' - 'sure you invoke the Eager delegate before inference.', + 'sure you invoke the Flex delegate before inference.', str(error.exception)) def testFloatTocoConverter(self): |