Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.lite.experimental.convert_op_hints_to_stubs

View source on GitHub

Converts a graphdef with LiteOp hints into stub operations.

Aliases:

  • 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.

Args:

  • 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).

Returns:

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

Raises:

  • ValueError: If both session and graph_def are provided.