Searched refs:set_algorithm (Results 1 – 14 of 14) sorted by relevance
70 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } in set_algorithm() function
219 best_result.set_algorithm(i); in AutotuneMatmul()224 algorithm_config.set_algorithm(best_result.algorithm()); in AutotuneMatmul()
495 best_result.set_algorithm(BiasAddGradGPUMode::kNative); in Compute()509 best_result.set_algorithm(BiasAddGradGPUMode::kReduction); in Compute()
445 algorithm_config.set_algorithm(0);473 best_result.set_algorithm(i);478 algorithm_config.set_algorithm(best_result.algorithm());
1542 output_algo_config->set_algorithm(algo_desc); in ComputeAndReturnAlgorithm()1801 best_result.set_algorithm(algo); in MaybeAutotune()1808 algo_config->set_algorithm(best_result.algorithm()); in MaybeAutotune()2098 algo_config->set_algorithm(algo_desc); in GetAlgorithm()
69 profile_result.set_algorithm(desc); in AutotuneConvImpl()
134 result.mutable_gemm()->set_algorithm(profile_result.algorithm()); in GetBestAlgorithm()179 result.mutable_failure()->mutable_reference_gemm()->set_algorithm( in GetBestAlgorithm()
157 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm() function172 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm() function
842 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm() function956 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm() function
870 output_profile_result->set_algorithm(blas::kDefaultBlasGemv); in DoBlasInternalImpl()904 output_profile_result->set_algorithm(blas::kDefaultBlasGemm); in DoBlasInternalImpl()1008 output_profile_result->set_algorithm(algorithm); in DoBlasInternalImpl()
2023 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()2133 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()2256 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()2365 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()4205 profile_result->set_algorithm(algo); in operator ()()4511 profile_result->set_algorithm(desc); in operator ()()5000 profile_result->set_algorithm(algo); in operator ()()
512 profile_result.set_algorithm( in GetProfileResultFromConvSolution()538 profile_result.set_algorithm({algo_id, false, algorithm.memory}); in GetProfileResultFromConvAlgoPerf()3081 profile_result->set_algorithm(algotype); in operator ()()
1107 optimization_params.set_algorithm(AutotuneAlgorithm::GRADIENT_DESCENT); in TEST()
2180 optimization_params.set_algorithm(algorithm); in Optimize()