diff options
author | 2018-05-17 13:12:51 -0700 | |
---|---|---|
committer | 2018-05-17 13:15:31 -0700 | |
commit | 7232a906caa549a108912999230ef0ec790b4dbd (patch) | |
tree | 991ad408025075f2636781d67e09b1f2e4d20b6f /tensorflow/contrib/lite/python/convert.py | |
parent | b39f1f5430f3b7fb2ae97f9dda7ad7cceca7250b (diff) |
Allows users to specify allow_custom_ops when calling tf.contrib.lite.toco_convert().
PiperOrigin-RevId: 197039477
Diffstat (limited to 'tensorflow/contrib/lite/python/convert.py')
-rw-r--r-- | tensorflow/contrib/lite/python/convert.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/python/convert.py b/tensorflow/contrib/lite/python/convert.py index c4200c879b..c0926d2f33 100644 --- a/tensorflow/contrib/lite/python/convert.py +++ b/tensorflow/contrib/lite/python/convert.py @@ -118,7 +118,8 @@ def toco_convert(input_data, input_format=lite_constants.TENSORFLOW_GRAPHDEF, output_format=lite_constants.TFLITE, quantized_input_stats=None, - drop_control_dependency=True): + drop_control_dependency=True, + allow_custom_ops=False): """Convert a model using TOCO from `input_format` to `output_format`. Typically this is to convert from TensorFlow GraphDef to TFLite, in which @@ -154,6 +155,7 @@ def toco_convert(input_data, toco.drop_control_dependency = drop_control_dependency model = _model_flags_pb2.ModelFlags() toco.inference_type = inference_type + toco.allow_custom_ops = allow_custom_ops for idx, input_tensor in enumerate(input_tensors): if input_tensor.dtype == _dtypes.float32: tflite_input_type = lite_constants.FLOAT |