Home
last modified time | relevance | path

Searched refs:DumpingEnabledForHloModule (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddump.h137 bool DumpingEnabledForHloModule(absl::string_view hlo_module_name,
139 inline bool DumpingEnabledForHloModule(const HloModule& module) { in DumpingEnabledForHloModule() function
140 return DumpingEnabledForHloModule(module.name(), in DumpingEnabledForHloModule()
H A Dservice.cc826 if (DumpingEnabledForHloModule(*module)) { in BuildExecutable()
845 CHECK(DumpingEnabledForHloModule(executable->module())); in BuildExecutable()
H A Ddump.cc686 bool DumpingEnabledForHloModule(string_view hlo_module_name, in DumpingEnabledForHloModule() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_compiler.cc879 if (!DumpingEnabledForHloModule(*module)) { in operator ()()
1282 DumpingEnabledForHloModule(cpu_executable->module())) { in CompileLegacyCpuExecutable()
1433 if (DumpingEnabledForHloModule(*module)) { in CompileAheadOfTime()
1543 if (!DumpingEnabledForHloModule(*module)) { in CompileAheadOfTime()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_compiler.cc1016 if (DumpingEnabledForHloModule(*hlo_module)) { in CompileModuleToLlvmIrImpl()
1140 DumpingEnabledForHloModule(debug_module ? debug_module->name() : "", in CompileToTargetBinary()
1386 if (DumpingEnabledForHloModule(*module) && in RunBackend()
1425 DumpingEnabledForHloModule(gpu_executable->module())) { in RunBackend()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dllvm_util.cc588 if (!DumpingEnabledForHloModule(hlo_module)) { in DumpIrIfEnabled()