Searched refs:set_algorithm (Results 1 – 18 of 18) sorted by relevance
473 result.mutable_conv()->set_algorithm(alg.algo_id()); in PickBestAlgorithmNoCacheCuda()546 reference_conv->set_algorithm(first_algorithm.algo_id()); in PickBestAlgorithmNoCacheCuda()684 result.mutable_conv()->set_algorithm(profile_result.algorithm().algo_id()); in PickBestAlgorithmNoCacheRocm()722 result.mutable_conv()->set_algorithm(alg.algo_id()); in PickBestAlgorithmNoCacheRocm()776 backend_config.set_algorithm(best_algo.conv().algorithm()); in RunOnInstruction()
113 result.mutable_gemm()->set_algorithm(algorithm); in DoUncachedGemmAutotune()155 result.mutable_failure()->mutable_reference_gemm()->set_algorithm( in DoUncachedGemmAutotune()
69 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } in set_algorithm() function
492 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()523 result.mutable_conv()->set_algorithm( in launch()547 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
574 best_result.set_algorithm(BiasAddGradGPUMode::kNative); in Compute()588 best_result.set_algorithm(BiasAddGradGPUMode::kReduction); in Compute()
271 algo->set_algorithm({selected_result->conv().algorithm(), in BestCudnnConvAlgorithm()
1021 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()1053 result.mutable_conv()->set_algorithm( in operator ()()1080 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1031 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()1065 result.mutable_conv()->set_algorithm( in operator ()()1089 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1188 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()1219 result.mutable_conv()->set_algorithm( in operator ()()1245 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1495 output_algo_config->set_algorithm(algo_desc); in ComputeAndReturnAlgorithm()1753 best_result.set_algorithm(algo); in MaybeAutoTune()1760 algo_config->set_algorithm(best_result.algorithm()); in MaybeAutoTune()2049 algo_config->set_algorithm(algo_desc); in GetAlgorithm()
1413 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()1464 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()1493 algorithm_config.set_algorithm(best_result.algorithm()); in Compute()1943 algorithm_config.set_algorithm(best_result.algorithm()); in Compute()
363 algorithm_config.set_algorithm(best_result.algorithm()); in launch()
355 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id());
805 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; }846 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; }
138 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm() function153 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm() function
1864 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()2021 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()3194 output_profile_result->set_algorithm(algorithm_desc); in DoConvolve()3325 output_profile_result->set_algorithm(algo_desc); in DoFusedConvolveImpl()
1823 output_profile_result->set_algorithm(blas::kDefaultBlasGemv); in DoBlasGemvWithProfilingImpl()1855 output_profile_result->set_algorithm(blas::kDefaultBlasGemm); in DoBlasGemmWithProfilingImpl()1974 output_profile_result->set_algorithm(algorithm); in DoBlasGemmWithAlgorithmImpl()
431 profile_result.set_algorithm({solution.solution_id, false}); in GetProfileResultFromConvSolution()2900 output_profile_result->set_algorithm(algotype); in DoConvolve()