From 39d1b3c7ee8a619b28832ac95e689ba34bac9500 Mon Sep 17 00:00:00 2001 From: Margaret Maynard-Reid Date: Sun, 24 Feb 2019 10:54:10 -0800 Subject: [PATCH 1/2] Updates description & example usage Made changes per issue #25844: - Updated description to mention tf.Keras model. - Added the missing line in the example usage for SavedModel and tf.keras model. --- tensorflow/lite/python/lite.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/lite/python/lite.py b/tensorflow/lite/python/lite.py index c8254518bd0..7ce41e89d50 100644 --- a/tensorflow/lite/python/lite.py +++ b/tensorflow/lite/python/lite.py @@ -291,7 +291,7 @@ class TFLiteConverterV2(object): class TFLiteConverter(object): """Convert a TensorFlow model into `output_format` using TOCO. - This is used to convert from a TensorFlow GraphDef or SavedModel into either a + This is used to convert from a TensorFlow GraphDef, SavedModel or tf.keras model into either a TFLite FlatBuffer or graph visualization. Attributes: @@ -370,10 +370,12 @@ class TFLiteConverter(object): # Converting a SavedModel. converter = lite.TFLiteConverter.from_saved_model(saved_model_dir) tflite_model = converter.convert() + open("converted_model.tflite", "wb").write(tflite_model) # Converting a tf.keras model. converter = lite.TFLiteConverter.from_keras_model_file(keras_model) tflite_model = converter.convert() + open("converted_model.tflite", "wb").write(tflite_model) ``` """ From 22e2ac61ccc690f7759cf2b8b33d3eddb2da4717 Mon Sep 17 00:00:00 2001 From: Margaret Maynard-Reid Date: Mon, 25 Feb 2019 15:45:45 -0800 Subject: [PATCH 2/2] Updated line to be within 80 characters --- tensorflow/lite/python/lite.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/lite/python/lite.py b/tensorflow/lite/python/lite.py index 7ce41e89d50..d4970e12634 100644 --- a/tensorflow/lite/python/lite.py +++ b/tensorflow/lite/python/lite.py @@ -291,8 +291,8 @@ class TFLiteConverterV2(object): class TFLiteConverter(object): """Convert a TensorFlow model into `output_format` using TOCO. - This is used to convert from a TensorFlow GraphDef, SavedModel or tf.keras model into either a - TFLite FlatBuffer or graph visualization. + This is used to convert from a TensorFlow GraphDef, SavedModel or tf.keras model + into either a TFLite FlatBuffer or graph visualization. Attributes: