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
|
||||
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
|
||||
|
Loading…
Reference in New Issue
Block a user