Searched refs:set_algorithm (Results 1 – 20 of 20) sorted by relevance
487 result.mutable_conv()->set_algorithm(alg.algo_id()); in PickBestAlgorithmNoCacheCuda()559 reference_conv->set_algorithm(first_algorithm.algo_id()); in PickBestAlgorithmNoCacheCuda()702 result.mutable_conv()->set_algorithm(profile_result.algorithm().algo_id()); in PickBestAlgorithmNoCacheRocm()741 result.mutable_conv()->set_algorithm(alg.algo_id()); in PickBestAlgorithmNoCacheRocm()795 backend_config.set_algorithm(best_algo.conv().algorithm()); in RunOnInstruction()
148 result.mutable_gemm()->set_algorithm(algorithm); in DoUncachedGemmAutotune()190 result.mutable_failure()->mutable_reference_gemm()->set_algorithm( in DoUncachedGemmAutotune()
1178 descriptor.backend_config.set_algorithm( in EmitConvolutionThunkFromMlir()
69 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } in set_algorithm() function
427 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()461 result.mutable_conv()->set_algorithm( in operator ()()482 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
548 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()582 result.mutable_conv()->set_algorithm( in launch()604 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
546 best_result.set_algorithm(BiasAddGradGPUMode::kNative); in Compute()560 best_result.set_algorithm(BiasAddGradGPUMode::kReduction); in Compute()
273 algo->set_algorithm({selected_result->conv().algorithm(), in BestCudnnConvAlgorithm()
1022 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()1059 result.mutable_conv()->set_algorithm( in operator ()()1080 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1034 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()1071 result.mutable_conv()->set_algorithm( in operator ()()1091 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1499 output_algo_config->set_algorithm(algo_desc); in ComputeAndReturnAlgorithm()1757 best_result.set_algorithm(algo); in MaybeAutoTune()1764 algo_config->set_algorithm(best_result.algorithm()); in MaybeAutoTune()2053 algo_config->set_algorithm(algo_desc); in GetAlgorithm()
1462 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()1501 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()1946 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()1978 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()
412 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id());
507 profile_result.set_algorithm( in GetProfileResultFromConvSolution()519 profile_result.set_algorithm( in GetProfileResultFromConvAlgoPerf()523 profile_result.set_algorithm( in GetProfileResultFromConvAlgoPerf()527 profile_result.set_algorithm( in GetProfileResultFromConvAlgoPerf()3066 output_profile_result->set_algorithm(algotype); in DoConvolve()
782 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm() function823 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm() function
168 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm() function183 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm() function
1877 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()2022 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()3166 output_profile_result->set_algorithm(algorithm_desc); in DoConvolve()3285 output_profile_result->set_algorithm(algo_desc); in DoFusedConvolveImpl()
2018 output_profile_result->set_algorithm(blas::kDefaultBlasGemv); in DoBlasInternalImpl()2050 output_profile_result->set_algorithm(blas::kDefaultBlasGemm); in DoBlasInternalImpl()2185 output_profile_result->set_algorithm(algorithm); in DoBlasInternalImpl()3733 output_profile_result->set_algorithm(algorithm->index()); in DoBlasInternalImpl()
946 optimization_params.set_algorithm(AutotuneAlgorithm::GRADIENT_DESCENT); in TEST()
1647 optimization_params.set_algorithm(algorithm); in Optimize()