Searched defs:gemm_config (Results 1 – 4 of 4) sorted by relevance
90 …std::unique_ptr<IClGemmKernelConfig> gemm_config = ClGemmReshapedOnlyRhsKernelConfigurationFactory… in select_default_gemm_config_reshaped_only_rhs() local125 …std::unique_ptr<IClGemmKernelConfig> gemm_config = ClGemmReshapedKernelConfigurationFactory::creat… in select_default_gemm_config_reshaped() local159 …std::unique_ptr<IClGemmKernelConfig> gemm_config = ClGemmNativeKernelConfigurationFactory::create(… in select_default_gemm_config_native() local
358 …auto gemm_config = select_default_gemm_config_reshaped_only_rhs(auto_heuristics::CommonQuery{ gpu_… in configure_reshaped_only_rhs_mmul() local443 …const auto gemm_config = select_default_gemm_config_reshaped(auto_heuristics::CommonQuery{ gpu_tar… in validate_reshaped() local492 …const auto gemm_config = select_default_gemm_config_reshaped_only_rhs(auto_heuristics::CommonQuery… in validate_reshaped_only_rhs() local541 …const auto gemm_config = select_default_gemm_config_reshaped_only_rhs(auto_heuristics::CommonQuery… in validate_reshaped_only_rhs_mmul() local
210 const HloInstruction* gemm, const GemmBackendConfig& gemm_config, in DoGemmAutotune()347 GemmBackendConfig gemm_config = in RunOnInstruction() local
145 GemmBackendConfig gemm_config; in HandleDot() local