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/
Dnvptx_helper.cc37 const HloModuleConfig& hlo_module_config) { in CantFindCudaMessage()
46 std::string GetLibdeviceDir(const HloModuleConfig& hlo_module_config) { in GetLibdeviceDir()
Dir_emitter_nested.cc43 IrEmitterNested::IrEmitterNested(const HloModuleConfig& hlo_module_config, in IrEmitterNested()
50 const HloModuleConfig& hlo_module_config, in Create()
Dgpu_conv_algorithm_picker.cc408 const HloModuleConfig& hlo_module_config = instr->GetModule()->config(); in ShouldInitConvData() local
415 const HloModuleConfig& hlo_module_config = instr->GetModule()->config(); in ShouldCheckConv() local
444 const auto& hlo_module_config = instr->GetModule()->config(); in AutotuneOneConvRunner() local
652 const HloModuleConfig& hlo_module_config = instr->GetModule()->config(); in PickBestAlgorithmNoCacheCuda() local
Dnvptx_compiler.cc381 se::CudaComputeCapability cc, const HloModuleConfig& hlo_module_config, in CompileGpuAsmOrGetCachedResult()
Delemental_ir_emitter.cc72 const HloModuleConfig& hlo_module_config, llvm::Module* module, in GpuElementalIrEmitter()
Dir_emitter.cc72 IrEmitter::IrEmitter(const HloModuleConfig& hlo_module_config, in IrEmitter()
Dir_emitter_unnested.cc284 const HloModuleConfig& hlo_module_config) { in ComputeMaxUnrollFactor()
307 const HloModuleConfig& hlo_module_config) { in ComputeMaxUnrollFactor()
482 IrEmitterUnnested::IrEmitterUnnested(const HloModuleConfig& hlo_module_config, in IrEmitterUnnested()
487 const HloModuleConfig& hlo_module_config, in Create()
/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()
338 const HloModuleConfig& hlo_module_config, in NVPTXTargetModuleLinker()
361 const HloModuleConfig& hlo_module_config) { in NVPTXGetTargetMachine()
372 const HloModuleConfig& hlo_module_config, in LinkAndOptimizeModule()
462 void NVPTXBackendInit(const HloModuleConfig& hlo_module_config) { in NVPTXBackendInit()
513 const HloModuleConfig& hlo_module_config, in CompileToPtx()
789 const HloModuleConfig& hlo_module_config, in AMDGPUTargetModuleLinker()
862 const HloModuleConfig& hlo_module_config) { in AMDGPUGetTargetMachine()
872 void AMDGPUBackendInit(const HloModuleConfig& hlo_module_config) { in AMDGPUBackendInit()
896 const HloModuleConfig& hlo_module_config, in CompileToHsaco()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc251 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in DotOpEmitter()
1241 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitNonBatchDotOperation()
1326 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in PotentiallyImplementedAsEigenMatmul()
1382 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitBatchDotOperation()
1509 const HloModuleConfig& hlo_module_config, in EmitDotOperation()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc240 const xla::HloModuleConfig& hlo_module_config) { in GetLibdeviceDir()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
Dkernel_ops_pattern.cc220 const xla::HloModuleConfig& hlo_module_config, llvm::Module* llvm_module) { in Emit()
276 xla::HloModuleConfig hlo_module_config; in Match() local