Fix TFLiteConverter2 API Documentation to read frozen_graphs.
PiperOrigin-RevId: 312785311 Change-Id: I6f5ec2dd5ee0d5796e3fd8c0c35fb50f78d56fab
This commit is contained in:
parent
0f178c3708
commit
4ce6280b48
@ -34,7 +34,7 @@ input_arrays = ['input_name']
|
|||||||
# A list of the names of the model's output tensors
|
# A list of the names of the model's output tensors
|
||||||
output_arrays = ['output_name']
|
output_arrays = ['output_name']
|
||||||
# Load and convert the frozen graph
|
# 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)
|
graph_def_file, input_arrays, output_arrays)
|
||||||
tflite_model = converter.convert()
|
tflite_model = converter.convert()
|
||||||
# Write the converted model to disk
|
# Write the converted model to disk
|
||||||
|
Loading…
Reference in New Issue
Block a user