| /external/tensorflow/tensorflow/python/ |
| D | pywrap_mlir.py | 27 def import_graphdef(graphdef, pass_pipeline, show_debug_info): argument
30 show_debug_info)
33 def import_function(concrete_function, pass_pipeline, show_debug_info): argument
38 pass_pipeline.encode('utf-8'), show_debug_info)
42 show_debug_info): argument
45 str(exported_names).encode('utf-8'), show_debug_info)
50 show_debug_info): argument
53 str(tags).encode('utf-8'), upgrade_legacy, show_debug_info)
58 show_debug_info): argument
62 show_debug_info)
[all …]
|
| D | mlir_wrapper.cc | 27 bool show_debug_info) { in PYBIND11_MODULE() argument
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()
52 const std::string &exported_names, bool show_debug_info) { in PYBIND11_MODULE() argument
56 saved_model_path, exported_names, show_debug_info, status.get()); in PYBIND11_MODULE()
63 bool upgrade_legacy, bool show_debug_info) { in PYBIND11_MODULE() argument
68 saved_model_path, tags, upgrade_legacy, show_debug_info, in PYBIND11_MODULE()
76 bool lift_variables, bool upgrade_legacy, bool show_debug_info) { in PYBIND11_MODULE() argument
82 show_debug_info, status.get()); in PYBIND11_MODULE()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/python/ |
| D | mlir.cc | 52 bool show_debug_info, TF_Status *status) { in RunPassPipelineOnModule() argument
69 return MlirModuleToString(module, show_debug_info); in RunPassPipelineOnModule()
76 bool show_debug_info, TF_Status *status) { in ImportGraphDef() argument
92 return RunPassPipelineOnModule(module->get(), pass_pipeline, show_debug_info, in ImportGraphDef()
98 bool show_debug_info, TFE_Context *tfe_context, in ImportFunction() argument
132 return RunPassPipelineOnModule(module->get(), pass_pipeline, show_debug_info, in ImportFunction()
138 bool show_debug_info, TF_Status *status) { in ExperimentalConvertSavedModelToMlir() argument
161 return MlirModuleToString(*module_or.ConsumeValueOrDie(), show_debug_info); in ExperimentalConvertSavedModelToMlir()
166 bool upgrade_legacy, bool show_debug_info, TF_Status *status) { in ExperimentalConvertSavedModelV1ToMlirLite() argument
182 return MlirModuleToString(*module_or.ValueOrDie(), show_debug_info); in ExperimentalConvertSavedModelV1ToMlirLite()
[all …]
|
| D | mlir.h | 35 bool show_debug_info, TF_Status *status);
44 bool show_debug_info, TFE_Context *context,
58 bool show_debug_info, TF_Status *status);
73 bool upgrade_legacy, bool show_debug_info, TF_Status *status);
87 bool lift_variables, bool upgrade_legacy, bool show_debug_info,
92 bool show_debug_info,
|
| /external/tensorflow/tensorflow/python/compiler/mlir/ |
| D | mlir.py | 28 show_debug_info=False): argument
50 return pywrap_mlir.import_graphdef(graph_def, pass_pipeline, show_debug_info)
56 show_debug_info=False): argument
95 show_debug_info)
|
| D | mlir_test.py | 56 mlir_module = mlir.convert_function(concrete_function, show_debug_info=True)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/ |
| D | common_v1.py | 51 show_debug_info=False, argument
110 upgrade_legacy, show_debug_info)
116 show_debug_info)
120 lift_variables, upgrade_legacy, show_debug_info)
124 show_debug_info)
|
| D | common.py | 43 def do_test(create_module_fn, exported_names=None, show_debug_info=False): argument
83 save_options = tf.saved_model.SaveOptions(save_debug_info=show_debug_info)
88 save_model_path, ','.join(exported_names), show_debug_info)
93 show_debug_info)
|
| D | debug_info.py | 42 common.do_test(TestModule, show_debug_info=True)
|
| /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…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.h | 137 std::string MlirModuleToString(mlir::ModuleOp m, bool show_debug_info = false);
|
| D | import_model.cc | 3956 std::string MlirModuleToString(mlir::ModuleOp module, bool show_debug_info) { in MlirModuleToString() argument
3958 if (show_debug_info) flags.enableDebugInfo(); in MlirModuleToString()
|
| /external/tensorflow/ |
| D | RELEASE.md | 147 * Adding show_debug_info to mlir.convert_graph_def and
|