Searched refs:debug_module (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 847 const HloModule* debug_module) { in CompileToTargetBinary() argument
851 [this, stream_exec, &module_config, debug_module]( in CompileToTargetBinary()
870 << (debug_module in CompileToTargetBinary()
872 FilenameFor(*debug_module, "", ""), "*") in CompileToTargetBinary()
878 stream_exec, relocatable, debug_module); in CompileToTargetBinary()
885 DumpingEnabledForHloModule(debug_module ? debug_module->name() : "", in CompileToTargetBinary()
889 if (debug_module) { in CompileToTargetBinary()
891 llvm_ir::DumpIrIfEnabled(*debug_module, *llvm_module, in CompileToTargetBinary()
895 llvm_ir::DumpIrIfEnabled(*debug_module, *llvm_module, in CompileToTargetBinary()
913 if (debug_module) { in CompileToTargetBinary()
[all …]
|
| D | nvptx_compiler.cc | 360 const HloModule* debug_module) { in CompileTargetBinary() argument
375 MaybeLoadLLVMFromFile(debug_module, llvm_module); in CompileTargetBinary()
384 if (!(debug_module && in CompileTargetBinary()
385 MaybeLoadPtxFromFile(module_config, debug_module, &ptx))) { in CompileTargetBinary()
|
| D | gpu_compiler.h | 87 const HloModule* debug_module) = 0;
103 const HloModule* debug_module);
|
| D | amdgpu_compiler.h | 45 const HloModule* debug_module) override;
|
| D | amdgpu_compiler.cc | 133 const HloModule* debug_module) { in CompileTargetBinary() argument
|
| D | nvptx_compiler.h | 59 const HloModule* debug_module) override;
|
| D | gpu_executable.h | 87 std::unique_ptr<HloModule> debug_module = nullptr; member
|
| D | gpu_executable.cc | 83 : Executable(std::move(params.debug_module), in GpuExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 63 const HloModule* debug_module) { in CompileTargetBinary() argument
|