aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/python
diff options
context:
space:
mode:
authorGravatar Nupur Garg <nupurgarg@google.com>2018-09-26 18:26:51 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-26 18:31:32 -0700
commit51a6118e5bd85935b1d9ec0e68b92f1f98d14982 (patch)
tree5b3d7b469bebd510b678a2054c04311e4879fd09 /tensorflow/contrib/lite/python
parent78ee51bb8c41d616c01674e405784a51e4ec98a5 (diff)
Automated rollback of commit 82af048bc8c3c044c98a27b1c4c27bb62d4e4a14
PiperOrigin-RevId: 214705311
Diffstat (limited to 'tensorflow/contrib/lite/python')
-rw-r--r--tensorflow/contrib/lite/python/convert.py6
-rw-r--r--tensorflow/contrib/lite/python/lite_test.py2
2 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/python/convert.py b/tensorflow/contrib/lite/python/convert.py
index 83d341c0b8..1f48a826d4 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_flex_ops = True
+ toco.allow_eager_ops = True
elif converter_mode == ConverterMode.TOCO_EXTENDED_ALL:
- toco.allow_flex_ops = True
- toco.force_flex_ops = True
+ toco.allow_eager_ops = True
+ toco.force_eager_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 89324e8a80..f112ed5cdd 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 Flex delegate before inference.',
+ 'sure you invoke the Eager delegate before inference.',
str(error.exception))