tf.lite.experimental.convert_op_hints_to_stubs

View source on GitHub

Converts a graphdef with LiteOp hints into stub operations.

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.lite.experimental.convert_op_hints_to_stubs

 
tf.lite.experimental.convert_op_hints_to_stubs(
    session=None, graph_def=None, write_callback=(lambda graph_def, comments: None)
)

This is used to prepare for toco conversion of complex intrinsic usages. Note: only one of session or graph_def should be used, not both.

session A TensorFlow session that contains the graph to convert.
graph_def A graph def that we should convert.
write_callback A function pointer that can be used to write intermediate steps of graph transformation (optional).

A new graphdef with all ops contained in OpHints being replaced by a single op call with the right parameters.

ValueError If both session and graph_def are provided.