Fix TFLiteConverter2 API Documentation to read frozen_graphs.

PiperOrigin-RevId: 312785311
Change-Id: I6f5ec2dd5ee0d5796e3fd8c0c35fb50f78d56fab
This commit is contained in:
Meghna Natraj 2020-05-21 19:46:14 -07:00 committed by TensorFlower Gardener
parent 0f178c3708
commit 4ce6280b48

View File

@ -34,7 +34,7 @@ input_arrays = ['input_name']
# A list of the names of the model's output tensors
output_arrays = ['output_name']
# Load and convert the frozen graph
converter = tf.lite.TFLiteConverter.from_frozen_graph(
converter = tf.compat.v1.lite.TFLiteConverter.from_frozen_graph(
graph_def_file, input_arrays, output_arrays)
tflite_model = converter.convert()
# Write the converted model to disk