Searched refs:debug_module (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 680 const HloModule* debug_module) { in CompileToTargetBinary() argument
684 [this, stream_exec, &module_config, debug_module]( in CompileToTargetBinary()
703 << (debug_module in CompileToTargetBinary()
705 FilenameFor(*debug_module, "", ""), "*") in CompileToTargetBinary()
711 stream_exec, relocatable, debug_module); in CompileToTargetBinary()
718 DumpingEnabledForHloModule(debug_module ? debug_module->name() : "", in CompileToTargetBinary()
722 if (debug_module) { in CompileToTargetBinary()
724 llvm_ir::DumpIrIfEnabled(*debug_module, *llvm_module, in CompileToTargetBinary()
728 llvm_ir::DumpIrIfEnabled(*debug_module, *llvm_module, in CompileToTargetBinary()
746 if (debug_module) { in CompileToTargetBinary()
[all …]
|
| D | gpu_compiler.h | 92 const HloModule* debug_module) = 0;
108 const HloModule* debug_module);
|
| D | nvptx_compiler.cc | 306 const HloModule* debug_module) { in CompileTargetBinary() argument
325 if (!(debug_module && in CompileTargetBinary()
326 MaybeLoadPtxFromFile(module_config, debug_module, &ptx))) { in CompileTargetBinary()
|
| D | amdgpu_compiler.h | 45 const HloModule* debug_module) override;
|
| D | amdgpu_compiler.cc | 119 const HloModule* debug_module) { in CompileTargetBinary() argument
|
| D | nvptx_compiler.h | 56 const HloModule* debug_module) override;
|
| D | gpu_executable.h | 81 std::unique_ptr<HloModule> debug_module = nullptr; member
|
| D | gpu_executable.cc | 58 : 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
|