ba167d161e
This takes a ConcreteFunction, collects a FunctionDef for the function and an associated FunctionDefLibrary, and imports the FunctionDef and FunctionDefLibrary via `ConvertFunctionToMlir`. Control rets/target nodes of the entry function are also now supported in `ConvertFunctionToMlir`. PiperOrigin-RevId: 331195841 Change-Id: Ib3a7264e90ca303ab7a850bf18c8d5e330063a4f
12 lines
408 B
Plaintext
12 lines
408 B
Plaintext
path: "tensorflow.mlir.experimental"
|
|
tf_module {
|
|
member_method {
|
|
name: "convert_function"
|
|
argspec: "args=[\'concrete_function\', \'pass_pipeline\'], varargs=None, keywords=None, defaults=[\'tf-standard-pipeline\'], "
|
|
}
|
|
member_method {
|
|
name: "convert_graph_def"
|
|
argspec: "args=[\'graph_def\', \'pass_pipeline\'], varargs=None, keywords=None, defaults=[\'tf-standard-pipeline\'], "
|
|
}
|
|
}
|