Searched refs:DumpHloModuleIfEnabled (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dump.h | 69 void DumpHloModuleIfEnabled(const HloModule& module, absl::string_view name);
70 void DumpHloModuleIfEnabled(const HloModule& module,
73 void DumpHloModuleIfEnabled(const HloModule& module,
|
| D | dump.cc | 409 void DumpHloModuleIfEnabled(const HloModule& module, string_view name) { in DumpHloModuleIfEnabled() function
416 void DumpHloModuleIfEnabled(const HloModule& module, in DumpHloModuleIfEnabled() function
426 void DumpHloModuleIfEnabled(const HloModule& module, in DumpHloModuleIfEnabled() function
|
| D | compile_only_service.cc | 112 DumpHloModuleIfEnabled(*hlo_module, "before_optimizations"); in CompileAheadOfTime()
|
| D | service.cc | 313 DumpHloModuleIfEnabled(*module, kBeforeOptimizationsDumpName); in BuildExecutables()
768 DumpHloModuleIfEnabled(*module, kBeforeOptimizationsDumpName); in BuildExecutable()
1116 DumpHloModuleIfEnabled(*module, kBeforeOptimizationsDumpName); in GetComputationGraphStats()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 729 DumpHloModuleIfEnabled(*module, *assignment, "after_optimizations"); in RunBackend()
931 DumpHloModuleIfEnabled(*module, *assignment, "after_optimizations"); in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 331 DumpHloModuleIfEnabled(*hlo_module, "before_optimizations"); in CreateHloModules()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 588 DumpHloModuleIfEnabled(*hlo_module, **buffer_assignment, in CompileModuleToLlvmIrImpl()
|