- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for MlirDumpConfig (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/dump_graph_test.cc
Node* node; TF_CHECK_OK(NodeBuilder("A", "NoOp").Finalize(&graph, &node)); string actual; StringWritableFile file(&actual); TF_ASSERT_OK(DumpTextualIRToFile( MlirDumpConfig().emit_dialect(MlirDumpConfig::Dialect::kTFG), graph, /*flib_def=*/nullptr, &file)); string expected_substr("tfg.graph"); ExpectHasSubstr(actual, expected_substr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_graph.cc
LOG(WARNING) << "Write failed: " << s; file = nullptr; } } } // The file being written to. WritableFile* file; }; } // namespace Status DumpTextualIRToFile(const MlirDumpConfig& config, const Graph& graph, const FunctionLibraryDefinition* flib_def, WritableFile* file) { WritableFileRawStream os(std::move(file)); mlir::MLIRContext context;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.cc
mlir_flags->tf_mlir_enable_multiple_local_cpu_devices = enable_mlir_multiple_local_cpu_devices; if (use_tfg_graph_dumper) { UseMlirForGraphDump(MlirDumpConfig{}.elide_large_attributes().emit_dialect( MlirDumpConfig::Dialect::kTFG)); } AllocateAndParseJitRtFlags(); } void ResetFlags() { delete build_ops_flags; delete mark_for_compilation_flags;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K bytes - Viewed (0)