Home
last modified time | relevance | path

Searched defs:hlo_module_config (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/
Dgpu_backend_lib.cc145 const HloModuleConfig& hlo_module_config, absl::string_view feature_str) { in GetTargetMachine()
339 const HloModuleConfig& hlo_module_config, in NVPTXTargetModuleLinker()
367 const HloModuleConfig& hlo_module_config) { in NVPTXGetTargetMachine()
378 const HloModuleConfig& hlo_module_config, in LinkAndOptimizeModule()
464 void NVPTXBackendInit(const HloModuleConfig& hlo_module_config) { in NVPTXBackendInit()
514 const HloModuleConfig& hlo_module_config, const string& libdevice_dir_path, in CompileToPtx()
787 const HloModuleConfig& hlo_module_config, in AMDGPUTargetModuleLinker()
856 const HloModuleConfig& hlo_module_config) { in AMDGPUGetTargetMachine()
866 void AMDGPUBackendInit(const HloModuleConfig& hlo_module_config) { in AMDGPUBackendInit()
888 const HloModuleConfig& hlo_module_config, const string& rocdl_dir_path) { in CompileToHsaco()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc70 const HloModuleConfig& hlo_module_config) { in PrintCantFindCudaMessage()
84 string GetLibdeviceDir(const HloModuleConfig& hlo_module_config) { in GetLibdeviceDir()
343 int cc_minor, const HloModuleConfig& hlo_module_config, bool relocatable) { in CompileGpuAsmOrGetCachedResult()
Dir_emitter_nested.cc38 IrEmitterNested::IrEmitterNested(const HloModuleConfig& hlo_module_config, in IrEmitterNested()
45 const HloModuleConfig& hlo_module_config, in Create()
Dgemm_algorithm_picker.cc67 const HloModuleConfig& hlo_module_config = gemm->GetModule()->config(); in DoUncachedGemmAutotune() local
Dstream_executor_util.cc226 se::GpuAsmOpts PtxOptsFromConfig(const HloModuleConfig& hlo_module_config) { in PtxOptsFromConfig()
Delemental_ir_emitter.cc73 const HloModuleConfig& hlo_module_config, llvm::Module* module, in GpuElementalIrEmitter()
Dir_emitter.cc79 IrEmitter::IrEmitter(const HloModuleConfig& hlo_module_config, in IrEmitter()
Dgpu_conv_algorithm_picker.cc389 const HloModuleConfig& hlo_module_config = instr->GetModule()->config(); in PickBestAlgorithmNoCacheCuda() local
Dir_emitter_unnested.cc320 const HloModuleConfig& hlo_module_config) { in ComputeMaxUnrollFactor()
347 const HloModuleConfig& hlo_module_config) { in ComputeMaxUnrollFactor()
531 IrEmitterUnnested::IrEmitterUnnested(const HloModuleConfig& hlo_module_config, in IrEmitterUnnested()
537 const HloModuleConfig& hlo_module_config, in Create()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc240 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in DotOpEmitter()
1044 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitNonBatchDotOperation()
1128 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitBatchDotOperation()
1237 const HloModuleConfig& hlo_module_config, in EmitDotOperation()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc235 const xla::HloModuleConfig& hlo_module_config) { in GetLibdeviceDir()