diff options
author | Yu-Cheng Ling <ycling@google.com> | 2018-09-26 14:57:43 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-26 15:03:20 -0700 |
commit | 82af048bc8c3c044c98a27b1c4c27bb62d4e4a14 (patch) | |
tree | d9285ea3c1ec60cdd1973c517975b61d14d14c1e /tensorflow/contrib/lite/python | |
parent | 83df298d2ed014d263570c7283322c982be556cc (diff) |
Rename TFLite Eager delegate -> Flex delegate
PiperOrigin-RevId: 214674717
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 1f48a826d4..83d341c0b8 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 f112ed5cdd..89324e8a80 100644 --- a/tensorflow/contrib/lite/python/lite_test.py +++ b/tensorflow/contrib/lite/python/lite_test.py @@ -421,7 +421,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)) |