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/convert.py | |
parent | 5f67bf69d3f53d1cd3bb86ebeeb03ea2bba5911b (diff) |
Rename TFLite Extended -> TFLite Flex
PiperOrigin-RevId: 214854303
Diffstat (limited to 'tensorflow/contrib/lite/python/convert.py')
-rw-r--r-- | tensorflow/contrib/lite/python/convert.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/python/convert.py b/tensorflow/contrib/lite/python/convert.py index 73a420c47b..613a1530f7 100644 --- a/tensorflow/contrib/lite/python/convert.py +++ b/tensorflow/contrib/lite/python/convert.py @@ -67,12 +67,12 @@ class ConverterMode(enum.Enum): # Convert model using TOCO such that only unsupported operations are # represented as TensorFlow ops. # WARNING: Experimental interface, subject to change. - TOCO_EXTENDED = "TOCO_EXTENDED" + TOCO_FLEX = "TOCO_FLEX" # Convert model using TOCO such that all operations are represented as # TensorFlow ops. # WARNING: Experimental interface, subject to change. - TOCO_EXTENDED_ALL = "TOCO_EXTENDED_ALL" + TOCO_FLEX_ALL = "TOCO_FLEX_ALL" def __str__(self): return self.value @@ -240,9 +240,9 @@ def build_toco_convert_protos(input_tensors, if dump_graphviz_dir: toco.dump_graphviz_dir = dump_graphviz_dir toco.dump_graphviz_include_video = dump_graphviz_video - if converter_mode == ConverterMode.TOCO_EXTENDED: + if converter_mode == ConverterMode.TOCO_FLEX: toco.allow_flex_ops = True - elif converter_mode == ConverterMode.TOCO_EXTENDED_ALL: + elif converter_mode == ConverterMode.TOCO_FLEX_ALL: toco.allow_flex_ops = True toco.force_flex_ops = True |