Home
last modified time | relevance | path

Searched refs:gemm_config (Results 1 – 7 of 7) sorted by relevance

/external/ComputeLibrary/src/runtime/CL/gemm_auto_heuristics/
DCLGEMMAutoHeuristics.cpp90 …std::unique_ptr<IClGemmKernelConfig> gemm_config = ClGemmReshapedOnlyRhsKernelConfigurationFactory… in select_default_gemm_config_reshaped_only_rhs() local
91 ARM_COMPUTE_ERROR_ON_NULLPTR(gemm_config.get()); in select_default_gemm_config_reshaped_only_rhs()
92 …std::tie(lhs_info, rhs_info) = gemm_config->configure(query.m, query.n, query.k, query.b, query.da… in select_default_gemm_config_reshaped_only_rhs()
125 …std::unique_ptr<IClGemmKernelConfig> gemm_config = ClGemmReshapedKernelConfigurationFactory::creat… in select_default_gemm_config_reshaped() local
126 ARM_COMPUTE_ERROR_ON_NULLPTR(gemm_config.get()); in select_default_gemm_config_reshaped()
127 …std::tie(lhs_info, rhs_info) = gemm_config->configure(query.m, query.n, query.k, query.b, query.da… in select_default_gemm_config_reshaped()
159 …std::unique_ptr<IClGemmKernelConfig> gemm_config = ClGemmNativeKernelConfigurationFactory::create(… in select_default_gemm_config_native() local
160 ARM_COMPUTE_ERROR_ON_NULLPTR(gemm_config.get()); in select_default_gemm_config_native()
161 …std::tie(lhs_info, rhs_info) = gemm_config->configure(query.m, query.n, query.k, query.b, query.da… in select_default_gemm_config_native()
/external/ComputeLibrary/examples/gemm_tuner/
DGemmTuner.py197 gemm_config: GEMMConfigT
218 gemm_param, strategy, gemm_config, measurement = benchmark_result
235 for gemm_param, strategy, gemm_config, measurement in self.get_record():
237 best_gc_set.append((gemm_config, measurement))
245 (gemm_config, measurement)
246 for gemm_config, measurement in best_gc_set[1:]
289 for gemm_param, strategy, gemm_config, measurement in self.get_record():
336 "GEMMConfig": str(res.gemm_config),
392 gemm_param, _, gemm_config, measurement = benchmark_result
393 self._gemm_config_dist[gemm_config].append((gemm_param, measurement))
[all …]
Dcl_gemm_benchmark.sh444 gemm_config=${array_configs[$array_configs_idx]}
446 …echo "Running shape[$array_shapes_idx]=$gemm_shape with config[$array_configs_idx]=$gemm_config" 1…
448 example_args="${gemm_shape},${gemm_config},--type=${DATA_TYPE},--tuner-mode=${TUNER_MODE}"
449 json_filename="${STRATEGY_OPTION}_${gemm_shape}_${gemm_config}_${DATA_TYPE}"
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_rewriter.cc145 GemmBackendConfig gemm_config; in HandleDot() local
146 gemm_config.set_alpha_real(1.0); in HandleDot()
147 gemm_config.set_alpha_imag(0.0); in HandleDot()
148 gemm_config.set_beta(0.0); in HandleDot()
149 *gemm_config.mutable_dot_dimension_numbers() = in HandleDot()
151 *gemm_config.mutable_precision_config() = instr->precision_config(); in HandleDot()
153 TF_RETURN_IF_ERROR(gemm_call->set_backend_config(gemm_config)); in HandleDot()
Dgemm_algorithm_picker.cc210 const HloInstruction* gemm, const GemmBackendConfig& gemm_config, in DoGemmAutotune() argument
221 gemm->shape(), gemm_config.SerializeAsString(), in DoGemmAutotune()
270 bool has_vector_bias = gemm_config.epilogue() == GemmBackendConfig::BIAS; in DoGemmAutotune()
347 GemmBackendConfig gemm_config = in RunOnInstruction() local
351 DoGemmAutotune(instr, gemm_config, allocator, stream)); in RunOnInstruction()
355 GemmBackendConfig updated_config = gemm_config; in RunOnInstruction()
362 return updated_config.SerializeAsString() != gemm_config.SerializeAsString(); in RunOnInstruction()
Dgpu_hlo_cost_analysis.cc29 TF_ASSIGN_OR_RETURN(auto gemm_config, in HandleCustomCall()
51 gemm_config.dot_dimension_numbers()); in HandleCustomCall()
/external/ComputeLibrary/src/gpu/cl/operators/
DClGemm.cpp358 …auto gemm_config = select_default_gemm_config_reshaped_only_rhs(auto_heuristics::CommonQuery{ gpu_… in configure_reshaped_only_rhs_mmul() local
359 lhs_info = gemm_config.lhs_info; in configure_reshaped_only_rhs_mmul()
360 rhs_info = gemm_config.rhs_info; in configure_reshaped_only_rhs_mmul()
443 …const auto gemm_config = select_default_gemm_config_reshaped(auto_heuristics::CommonQuery{ gpu_tar… in validate_reshaped() local
444 lhs_info = gemm_config.lhs_info; in validate_reshaped()
445 rhs_info = gemm_config.rhs_info; in validate_reshaped()
492 …const auto gemm_config = select_default_gemm_config_reshaped_only_rhs(auto_heuristics::CommonQuery… in validate_reshaped_only_rhs() local
493 lhs_info = gemm_config.lhs_info; in validate_reshaped_only_rhs()
494 rhs_info = gemm_config.rhs_info; in validate_reshaped_only_rhs()
541 …const auto gemm_config = select_default_gemm_config_reshaped_only_rhs(auto_heuristics::CommonQuery… in validate_reshaped_only_rhs_mmul() local
[all …]