Searched defs:hlo_module_config (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/ |
| D | nvptx_backend_lib.cc | 168 const HloModuleConfig& hlo_module_config) { in GetTargetMachine()
330 const HloModuleConfig& hlo_module_config, in CompileModuleToPtx()
437 void GPUBackendInit(const HloModuleConfig& hlo_module_config) { in GPUBackendInit()
484 const HloModuleConfig& hlo_module_config, in CompileToPtx()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 122 const HloModuleConfig& hlo_module_config) { in GetCudaRootCandidates()
140 const HloModuleConfig& hlo_module_config) { in PrintCantFindCudaMessage()
153 string GetLibdeviceDir(const HloModuleConfig& hlo_module_config) { in GetLibdeviceDir()
541 const HloModuleConfig& hlo_module_config) { in CompilePtx()
805 const HloModuleConfig& hlo_module_config) { in CompilePtxOrGetCachedResult()
|
| D | ir_emitter_nested.cc | 38 IrEmitterNested::IrEmitterNested(const HloModuleConfig& hlo_module_config, in IrEmitterNested()
|
| D | elemental_ir_emitter.cc | 71 const HloModuleConfig& hlo_module_config, llvm::Module* module, in GpuElementalIrEmitter()
|
| D | ir_emitter.cc | 57 IrEmitter::IrEmitter(const HloModuleConfig& hlo_module_config, in IrEmitter()
|
| D | ir_emitter_unnested.cc | 174 IrEmitterUnnested::IrEmitterUnnested(const HloModuleConfig& hlo_module_config, in IrEmitterUnnested()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.h | 37 ElementalIrEmitter(const HloModuleConfig& hlo_module_config, in ElementalIrEmitter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 212 const HloModuleConfig& hlo_module_config, in DotOpEmitter()
850 const HloModuleConfig& hlo_module_config, in EmitNonBatchDotOperation()
932 const HloModuleConfig& hlo_module_config, in EmitBatchDotOperation()
1037 const HloModuleConfig& hlo_module_config, in EmitDotOperation()
|