| /external/tensorflow/tensorflow/python/ |
| D | pywrap_mlir.py | 27 def import_graphdef(graphdef, pass_pipeline, show_debug_info): argument
29 str(graphdef).encode('utf-8'), pass_pipeline.encode('utf-8'),
33 def import_function(concrete_function, pass_pipeline, show_debug_info): argument
38 pass_pipeline.encode('utf-8'), show_debug_info)
65 def experimental_run_pass_pipeline(mlir_txt, pass_pipeline, show_debug_info): argument
67 mlir_txt.encode('utf-8'), pass_pipeline.encode('utf-8'), show_debug_info)
|
| D | mlir_wrapper.cc | 26 [](const std::string &graphdef, const std::string &pass_pipeline, in PYBIND11_MODULE()
31 graphdef, pass_pipeline, show_debug_info, status.get()); in PYBIND11_MODULE()
38 const std::string &pass_pipeline, bool show_debug_info) { in PYBIND11_MODULE() argument
45 functiondef, pass_pipeline, show_debug_info, ctxt, status.get()); in PYBIND11_MODULE()
88 [](const std::string &mlir_txt, const std::string &pass_pipeline, in PYBIND11_MODULE()
93 mlir_txt, pass_pipeline, show_debug_info, status.get()); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/compiler/mlir/ |
| D | mlir.py | 27 pass_pipeline='tf-standard-pipeline', argument
50 return pywrap_mlir.import_graphdef(graph_def, pass_pipeline, show_debug_info)
55 pass_pipeline='tf-standard-pipeline', argument
94 return pywrap_mlir.import_function(concrete_function, pass_pipeline,
|
| D | mlir_test.py | 85 mlir_module = mlir.convert_function(concrete_function, pass_pipeline='')
|
| /external/tensorflow/tensorflow/compiler/mlir/python/ |
| D | mlir.cc | 51 const std::string &pass_pipeline, in RunPassPipelineOnModule() argument
53 if (!pass_pipeline.empty()) { in RunPassPipelineOnModule()
57 if (failed(mlir::parsePassPipeline(pass_pipeline, pm, error_stream))) { in RunPassPipelineOnModule()
75 const std::string &pass_pipeline, in ImportGraphDef() argument
92 return RunPassPipelineOnModule(module->get(), pass_pipeline, show_debug_info, in ImportGraphDef()
97 const std::string &pass_pipeline, in ImportFunction() argument
132 return RunPassPipelineOnModule(module->get(), pass_pipeline, show_debug_info, in ImportFunction()
238 const std::string &pass_pipeline, in ExperimentalRunPassPipeline() argument
259 if (failed(mlir::parsePassPipeline(pass_pipeline, pm, error_stream))) { in ExperimentalRunPassPipeline()
|
| D | mlir.h | 34 const std::string &pass_pipeline,
43 const std::string &pass_pipeline,
91 const std::string &pass_pipeline,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.mlir.experimental.pbtxt | 5 …argspec: "args=[\'concrete_function\', \'pass_pipeline\', \'show_debug_info\'], varargs=None, keyw…
9 …argspec: "args=[\'graph_def\', \'pass_pipeline\', \'show_debug_info\'], varargs=None, keywords=Non…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.mlir.experimental.pbtxt | 5 …argspec: "args=[\'concrete_function\', \'pass_pipeline\', \'show_debug_info\'], varargs=None, keyw…
9 …argspec: "args=[\'graph_def\', \'pass_pipeline\', \'show_debug_info\'], varargs=None, keywords=Non…
|