Home
last modified time | relevance | path

Searched refs:hlo_module_config_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Delemental_ir_emitter.h35 hlo_module_config_(module_config), in CpuElementalIrEmitter()
52 return hlo_module_config_.debug_options().xla_cpu_enable_fast_min_max(); in fast_min_max()
55 const HloModuleConfig& hlo_module_config_; variable
Ddot_op_emitter.cc192 return options::LlvmIrGemvTilingFactor(hlo_module_config_) in GetGemvTilingFactor()
203 return options::LlvmIrGemmTileSize(hlo_module_config_) in GetGemmTileSize()
230 const HloModuleConfig& hlo_module_config_; member in xla::cpu::__anonf3b678a30111::DotOpEmitter
251 hlo_module_config_(hlo_module_config), in DotOpEmitter()
377 /*rhs=*/rhs, /*result=*/target, b_, hlo_module_config_); in EmitTiledLlvmIrGemm()
480 /*result=*/result_op, b_, hlo_module_config_); in EmitTiledLlvmIrGemv()
490 /*result=*/result_op, b_, hlo_module_config_); in EmitTiledLlvmIrGemv()
527 switch (GetDotImplementationStrategy(hlo_module_config_, dot_info_, in Emit()
739 bool multi_threaded = ShouldUseMultiThreadedEigen(hlo_module_config_); in EmitCallToRuntime()
740 bool use_mkl_dnn = hlo_module_config_.debug_options().xla_cpu_use_mkl_dnn(); in EmitCallToRuntime()
Dir_emitter.cc106 hlo_module_config_(hlo_module.config()), in IrEmitter()
110 b_.setFastMathFlags(llvm_ir::GetCpuFastMathFlags(hlo_module_config_)); in IrEmitter()
220 hlo_module_config_, module_, &b_, in InitializeIrFunction()
867 hlo_module_config_, target_machine_features_); in HandleDot()
959 hlo_module_config_.debug_options().xla_cpu_multi_thread_eigen(); in HandleConvolution()
961 hlo_module_config_.debug_options().xla_cpu_use_mkl_dnn(); in HandleConvolution()
1043 hlo_module_config_.debug_options().xla_cpu_multi_thread_eigen(); in HandleFft()
1202 if (hlo_module_config_.replica_count() == 1) { in HandleAllReduce()
1799 if (!options::VectorizedReduceDisabled(hlo_module_config_)) { in HandleReduce()
2067 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion()
[all …]
Dir_emitter.h586 const HloModuleConfig& hlo_module_config_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.h102 return hlo_module_config_.debug_options().xla_gpu_enable_fast_min_max(); in fast_min_max()
133 const HloModuleConfig& hlo_module_config_; variable
Dir_emitter.cc87 hlo_module_config_(hlo_module_config) { in IrEmitter()
99 *hlo, GpuElementalIrEmitter(hlo_module_config_, module_, &b_, in DefaultAction()
233 IrEmitterNested::Create(hlo_module_config_, nested_computation, in EmitCallToNestedComputation()
580 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in HandleFusion()
Delemental_ir_emitter.cc76 hlo_module_config_(hlo_module_config), in GpuElementalIrEmitter()
169 if (hlo_module_config_.debug_options().xla_gpu_enable_fast_min_max() && in EmitFloatBinaryOp()
Dir_emitter.h180 const HloModuleConfig& hlo_module_config_; variable
Dir_emitter_unnested.cc729 unroll_factor = ComputeMaxUnrollFactor(input.op, hlo_module_config_); in EmitUsingElementalIrEmitter()
1794 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in EmitLoopFusionFromMlir()
1815 unroll_factor = ComputeMaxUnrollFactor(fusion, hlo_module_config_); in EmitLoopFusionFromMlir()
1899 hlo_module_config_, ir_emitter_context_->llvm_module(), &b_, in HandleFusion()
1916 ComputeMaxUnrollFactor(fusion_op, hlo_module_config_); in HandleFusion()
1943 hlo_module_config_, ir_emitter_context_->llvm_module(), &b_, in HandleFusion()
2031 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in HandleFusion()
2168 if (hlo_module_config_.debug_options().xla_gpu_deterministic_ops()) { in AssertNonDeterminismIsOkay()
3104 int64 replica_count = hlo_module_config_.replica_count(); in EmitNcclThunkFromMlir()
3727 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in BuildFusedInitializerThunkForMlir()
[all …]