Searched refs:DumpHloModuleIfEnabled (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dump.h | 55 void DumpHloModuleIfEnabled(const HloModule& module, absl::string_view name);
56 void DumpHloModuleIfEnabled(const HloModule& module,
59 void DumpHloModuleIfEnabled(const HloModule& module,
|
| D | dump.cc | 265 void DumpHloModuleIfEnabled(const HloModule& module, string_view name) { in DumpHloModuleIfEnabled() function
272 void DumpHloModuleIfEnabled(const HloModule& module, in DumpHloModuleIfEnabled() function
281 void DumpHloModuleIfEnabled(const HloModule& module, in DumpHloModuleIfEnabled() function
|
| D | compile_only_service.cc | 91 DumpHloModuleIfEnabled(*hlo_module, "before_optimizations"); in CompileAheadOfTime()
|
| D | service.cc | 373 DumpHloModuleIfEnabled(*module, kBeforeOptimizationsDumpName); in BuildExecutables()
793 DumpHloModuleIfEnabled(*module, kBeforeOptimizationsDumpName); in BuildExecutable()
1139 DumpHloModuleIfEnabled(*module, kBeforeOptimizationsDumpName); in GetComputationGraphStats()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 656 DumpHloModuleIfEnabled(*module, *assignment, "after_optimizations"); in RunBackend()
841 DumpHloModuleIfEnabled(*module, *assignment, "after_optimizations"); in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 679 DumpHloModuleIfEnabled(*module, *buffer_assignment, "after_optimizations"); in RunBackend()
|