experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
cherrypick_339370490
STT-tensorflow/tensorflow/python/compiler/mlir
History
Andy Ly ba167d161e Add python wrapper mlir.experimental.convert_function for importing ConcreteFunctions into TF MLIR. 
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
2020-09-11 12:21:46 -07:00
..
BUILD Add python wrapper mlir.experimental.convert_function for importing ConcreteFunctions into TF MLIR. 2020-09-11 12:21:46 -07:00
mlir_test.py Add python wrapper mlir.experimental.convert_function for importing ConcreteFunctions into TF MLIR. 2020-09-11 12:21:46 -07:00
mlir.py Add python wrapper mlir.experimental.convert_function for importing ConcreteFunctions into TF MLIR. 2020-09-11 12:21:46 -07:00