Home
last modified time | relevance | path

Searched refs:hlo_module_config (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/
Dnvptx_backend_lib.cc168 const HloModuleConfig& hlo_module_config) { in GetTargetMachine() argument
185 switch (hlo_module_config.debug_options().xla_backend_optimization_level()) { in GetTargetMachine()
330 const HloModuleConfig& hlo_module_config, in CompileModuleToPtx() argument
347 hlo_module_config.debug_options().xla_gpu_ftz()); in CompileModuleToPtx()
350 if (hlo_module_config.debug_options().xla_gpu_ftz()) { in CompileModuleToPtx()
375 target_triple, GetSmName(compute_capability), hlo_module_config); in CompileModuleToPtx()
389 hlo_module_config.debug_options().xla_backend_optimization_level(); in CompileModuleToPtx()
437 void GPUBackendInit(const HloModuleConfig& hlo_module_config) { in GPUBackendInit() argument
466 llvm_ir::InitializeLLVMCommandLineOptions(hlo_module_config); in GPUBackendInit()
484 const HloModuleConfig& hlo_module_config, in CompileToPtx() argument
[all …]
Dnvptx_backend_lib.h41 const HloModuleConfig& hlo_module_config,
DBUILD33 "//tensorflow/compiler/xla/service:hlo_module_config",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc122 const HloModuleConfig& hlo_module_config) { in GetCudaRootCandidates() argument
131 hlo_module_config.debug_options().xla_gpu_cuda_data_dir(); in GetCudaRootCandidates()
140 const HloModuleConfig& hlo_module_config) { in PrintCantFindCudaMessage() argument
143 for (const auto& dir : GetCudaRootCandidates(hlo_module_config)) { in PrintCantFindCudaMessage()
153 string GetLibdeviceDir(const HloModuleConfig& hlo_module_config) { in GetLibdeviceDir() argument
154 const auto& candidate_dirs = GetCudaRootCandidates(hlo_module_config); in GetLibdeviceDir()
168 hlo_module_config); in GetLibdeviceDir()
541 const HloModuleConfig& hlo_module_config) { in CompilePtx() argument
545 for (const string& cuda_root : GetCudaRootCandidates(hlo_module_config)) { in CompilePtx()
586 if (hlo_module_config.debug_options().xla_gpu_disable_ptxas_optimizations()) { in CompilePtx()
[all …]
Dir_emitter_nested.cc38 IrEmitterNested::IrEmitterNested(const HloModuleConfig& hlo_module_config, in IrEmitterNested() argument
41 : IrEmitter(hlo_module_config, ir_emitter_context, /*is_nested=*/true) { in IrEmitterNested()
Dir_emitter_nested.h45 IrEmitterNested(const HloModuleConfig& hlo_module_config,
Delemental_ir_emitter.cc71 const HloModuleConfig& hlo_module_config, llvm::Module* module, in GpuElementalIrEmitter() argument
73 : ElementalIrEmitter(hlo_module_config, module, b), in GpuElementalIrEmitter()
74 hlo_module_config_(hlo_module_config), in GpuElementalIrEmitter()
Dnvptx_compiler.h102 const HloModuleConfig& hlo_module_config);
Delemental_ir_emitter.h45 GpuElementalIrEmitter(const HloModuleConfig& hlo_module_config,
Dir_emitter.h113 explicit IrEmitter(const HloModuleConfig& hlo_module_config,
Dir_emitter.cc57 IrEmitter::IrEmitter(const HloModuleConfig& hlo_module_config, in IrEmitter() argument
65 hlo_module_config_(hlo_module_config) { in IrEmitter()
Dir_emitter_unnested.h146 IrEmitterUnnested(const HloModuleConfig& hlo_module_config,
DBUILD242 "//tensorflow/compiler/xla/service:hlo_module_config",
686 "//tensorflow/compiler/xla/service:hlo_module_config",
Dir_emitter_unnested.cc174 IrEmitterUnnested::IrEmitterUnnested(const HloModuleConfig& hlo_module_config, in IrEmitterUnnested() argument
177 : IrEmitter(hlo_module_config, ir_emitter_context, /*is_nested=*/false), in IrEmitterUnnested()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
DBUILD89 "//tensorflow/compiler/xla/service:hlo_module_config",
126 "//tensorflow/compiler/xla/service:hlo_module_config",
174 "//tensorflow/compiler/xla/service:hlo_module_config",
188 "//tensorflow/compiler/xla/service:hlo_module_config",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc116 const HloModuleConfig& hlo_module_config,
212 const HloModuleConfig& hlo_module_config, in DotOpEmitter() argument
222 hlo_module_config_(hlo_module_config), in DotOpEmitter()
850 const HloModuleConfig& hlo_module_config, in EmitNonBatchDotOperation() argument
857 executable_run_options_value, b, hlo_module_config, in EmitNonBatchDotOperation()
932 const HloModuleConfig& hlo_module_config, in EmitBatchDotOperation() argument
990 executable_run_options_value, b, hlo_module_config, in EmitBatchDotOperation()
1037 const HloModuleConfig& hlo_module_config, in EmitDotOperation() argument
1047 hlo_module_config, target_machine_features); in EmitDotOperation()
1053 hlo_module_config, target_machine_features); in EmitDotOperation()
Ddot_op_emitter.h67 const HloModuleConfig& hlo_module_config,
DBUILD288 "//tensorflow/compiler/xla/service:hlo_module_config",
349 "//tensorflow/compiler/xla/service:hlo_module_config",
381 "//tensorflow/compiler/xla/service:hlo_module_config",
409 "//tensorflow/compiler/xla/service:hlo_module_config",
929 "//tensorflow/compiler/xla/service:hlo_module_config",
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
DBUILD46 "//tensorflow/compiler/xla/service:hlo_module_config",
91 "//tensorflow/compiler/xla/service:hlo_module_config",
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h37 ElementalIrEmitter(const HloModuleConfig& hlo_module_config, in ElementalIrEmitter() argument
39 : b_(b), module_(module), hlo_module_config_(hlo_module_config) {} in ElementalIrEmitter()
DBUILD350 ":hlo_module_config",
732 ":hlo_module_config",
773 ":hlo_module_config",
940 ":hlo_module_config",
1487 ":hlo_module_config",
2582 ":hlo_module_config",
2730 ":hlo_module_config",
3184 ":hlo_module_config",
3225 name = "hlo_module_config",
3226 srcs = ["hlo_module_config.cc"],
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
DBUILD72 "//tensorflow/compiler/xla/service:hlo_module_config",