tf.mlir.experimental.convert_graph_def

View source on GitHub

Import a GraphDef and convert it to a textual MLIR module.

tf.mlir.experimental.convert_graph_def(
    graph_def, pass_pipeline='tf-standard-pipeline'
)

Args:

  • graph_def: An object of type graph_pb2.GraphDef or a textual proto representation of a valid GraphDef.
  • pass_pipeline: A textual description of an MLIR Pass Pipeline to run on the module, see MLIR documentation for the textual pass pipeline syntax.

Returns:

A textual representation of the MLIR module corresponding to the graphdef. Raises a RuntimeError on error.