Home
last modified time | relevance | path

Searched refs:first_algorithm (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_algorithm_picker.cc82 absl::optional<se::blas::AlgorithmType> first_algorithm; in DoUncachedGemmAutotune() local
137 if (!first_algorithm) { in DoUncachedGemmAutotune()
142 first_algorithm.emplace(algorithm); in DoUncachedGemmAutotune()
156 *first_algorithm); in DoUncachedGemmAutotune()
Dgpu_conv_algorithm_picker.cc413 AlgorithmDesc first_algorithm; in PickBestAlgorithmNoCacheCuda() local
522 LOG(ERROR) << "Unable to compare " << AlgorithmToString(first_algorithm) in PickBestAlgorithmNoCacheCuda()
536 << AlgorithmToString(first_algorithm) << " vs " in PickBestAlgorithmNoCacheCuda()
546 reference_conv->set_algorithm(first_algorithm.algo_id()); in PickBestAlgorithmNoCacheCuda()
548 first_algorithm.tensor_ops_enabled()); in PickBestAlgorithmNoCacheCuda()
558 first_algorithm = alg; in PickBestAlgorithmNoCacheCuda()