Home
last modified time | relevance | path

Searched refs:DumpMlirOpToFile (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddump_mlir_util_test.cc35 std::string filepath = DumpMlirOpToFile("module", module_ref.get()); in TEST()
45 std::string filepath = DumpMlirOpToFile("module", module_ref.get()); in TEST()
62 std::string filepath = DumpMlirOpToFile("module", module_ref.get()); in TEST()
Dcompile_mlir_util.cc350 tensorflow::DumpMlirOpToFile("legalize_hlo_before", module_op); in LegalizeToHlo()
370 tensorflow::DumpMlirOpToFile("legalize_hlo_after", module_op); in LegalizeToHlo()
413 tensorflow::DumpMlirOpToFile("compile_mlir_shape_refiner", module_op); in CompileMlirSetup()
429 tensorflow::DumpMlirOpToFile("build_hlo_tf_before", module_op); in BuildHloFromTf()
441 tensorflow::DumpMlirOpToFile("build_hlo_tf_after", module_op); in BuildHloFromTf()
607 tensorflow::DumpMlirOpToFile("compile_graph_setup_before", module_op); in CompileGraphSetup()
611 tensorflow::DumpMlirOpToFile("compile_graph_setup_after", module_op); in CompileGraphSetup()
Ddump_mlir_util.h51 std::string DumpMlirOpToFile(llvm::StringRef name, mlir::Operation* op,
Ddump_mlir_util.cc156 std::string DumpMlirOpToFile(llvm::StringRef name, mlir::Operation* op, in DumpMlirOpToFile() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dbridge.cc63 tensorflow::DumpMlirOpToFile("tpu_bridge_before", module); in RunTPUBridge()
79 tensorflow::DumpMlirOpToFile("tpu_bridge_after", module); in RunTPUBridge()
172 tensorflow::DumpMlirOpToFile("standard_pipeline_before", module); in RunBridgeWithStandardPipeline()
183 tensorflow::DumpMlirOpToFile("standard_pipeline_after", module); in RunBridgeWithStandardPipeline()