experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Throw error for using tflite_convert with the command line.

Browse Source 
PiperOrigin-RevId: 236148909
This commit is contained in:
Nupur Garg 2019-02-28 10:32:18 -08:00 committed by TensorFlower Gardener
parent 18285855e7
commit 7db477e1a1
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/python/tflite_convert.py
View File

@ -25,6 +25,7 @@ import sys
from tensorflow.lite.python import lite from tensorflow.lite.python import lite
from tensorflow.lite.python import lite_constants from tensorflow.lite.python import lite_constants
from tensorflow.lite.toco import toco_flags_pb2 as _toco_flags_pb2 from tensorflow.lite.toco import toco_flags_pb2 as _toco_flags_pb2
from tensorflow.python import tf2
from tensorflow.python.platform import app from tensorflow.python.platform import app
@ -258,6 +259,11 @@ def _check_flags(flags, unparsed):
def run_main(_): def run_main(_):
"""Main in toco_convert.py.""" """Main in toco_convert.py."""
if tf2.enabled():
raise ValueError("tflite_convert is currently unsupported in 2.0. "
"Please use the Python API "
"tf.lite.TFLiteConverter.from_concrete_function().")
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description=("Command line tool to run TensorFlow Lite Optimizing " description=("Command line tool to run TensorFlow Lite Optimizing "
"Converter (TOCO).")) "Converter (TOCO)."))