diff options
author | 2018-09-27 16:19:09 -0700 | |
---|---|---|
committer | 2018-09-27 16:33:31 -0700 | |
commit | 2330933ddd0b29ad206e351c9120e621cdaf6312 (patch) | |
tree | c617ed415b787d276619fcd19f32b6aa1e133453 /tensorflow/contrib/lite/testing | |
parent | 5f67bf69d3f53d1cd3bb86ebeeb03ea2bba5911b (diff) |
Rename TFLite Extended -> TFLite Flex
PiperOrigin-RevId: 214854303
Diffstat (limited to 'tensorflow/contrib/lite/testing')
-rw-r--r-- | tensorflow/contrib/lite/testing/generate_examples.py | 10 | ||||
-rw-r--r-- | tensorflow/contrib/lite/testing/model_coverage/model_coverage_lib_test.py | 2 |
2 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/contrib/lite/testing/generate_examples.py b/tensorflow/contrib/lite/testing/generate_examples.py index 53bd88d087..18036fac6f 100644 --- a/tensorflow/contrib/lite/testing/generate_examples.py +++ b/tensorflow/contrib/lite/testing/generate_examples.py @@ -81,9 +81,9 @@ parser.add_argument( action="store_true", help="Include intermediate graphdefs in the output zip files.") parser.add_argument( - "--run_with_extended", + "--run_with_flex", action="store_true", - help="Whether the TFLite Extended converter is being used.") + help="Whether the TFLite Flex converter is being used.") RANDOM_SEED = 342 TEST_INPUT_DEPTH = 3 @@ -339,10 +339,10 @@ def toco_convert(graph_def_str, input_tensors, output_tensors, graphdef_file.flush() # TODO(aselle): Switch this to subprocess at some point. - if "pb2lite" in bin_path and FLAGS.run_with_extended: + if "pb2lite" in bin_path and FLAGS.run_with_flex: opts = ("--input_arrays={0} --output_arrays={1}".format( ",".join(input_arrays), ",".join(output_tensors))) - elif FLAGS.run_with_extended: + elif FLAGS.run_with_flex: opts += " --allow_flex_ops --force_flex_ops" cmd = ("%s --input_file=%s --output_file=%s %s > %s 2>&1" % (bin_path, graphdef_file.name, output_file.name, opts, @@ -3333,7 +3333,7 @@ def main(unused_args): # list of valid conversion modes is defined in # generated_test_conversion_modes() in build_def.bzl. test_function = ("make_%s_tests" % (out.replace(".zip", "").replace( - "pb2lite", "").replace("toco-extended", "").rstrip("_"))) + "pb2lite", "").replace("toco-flex", "").rstrip("_"))) if test_function not in globals(): raise RuntimeError("Can't find a test function to create %r. Tried %r" % (out, test_function)) diff --git a/tensorflow/contrib/lite/testing/model_coverage/model_coverage_lib_test.py b/tensorflow/contrib/lite/testing/model_coverage/model_coverage_lib_test.py index 5f3355e734..1498f86c6f 100644 --- a/tensorflow/contrib/lite/testing/model_coverage/model_coverage_lib_test.py +++ b/tensorflow/contrib/lite/testing/model_coverage/model_coverage_lib_test.py @@ -123,7 +123,7 @@ class EvaluateKerasModel(test.TestCase): keras_file = self._saveKerasModel(model) model_coverage.test_keras_model( - keras_file, converter_mode=lite.ConverterMode.TOCO_EXTENDED) + keras_file, converter_mode=lite.ConverterMode.TOCO_FLEX) if __name__ == '__main__': |