Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_algorithm_picker.cc116 std::vector<AutotuneResult> profile_results; in DoUncachedGemmAutotune() local
146 profile_results.emplace_back(); in DoUncachedGemmAutotune()
147 AutotuneResult& result = profile_results.back(); in DoUncachedGemmAutotune()
197 for (const AutotuneResult& profile : profile_results) { in DoUncachedGemmAutotune()
217 profile_results, in DoUncachedGemmAutotune()
222 if (best_result != profile_results.end() && !has_failure(*best_result)) { in DoUncachedGemmAutotune()
Dgpu_conv_algorithm_picker.cc429 std::vector<AutotuneResult> profile_results; in PickBestAlgorithmNoCacheCuda() local
485 profile_results.emplace_back(); in PickBestAlgorithmNoCacheCuda()
486 AutotuneResult& result = profile_results.back(); in PickBestAlgorithmNoCacheCuda()
590 for (const auto& profile : profile_results) { in PickBestAlgorithmNoCacheCuda()
608 for (const auto& result : profile_results) { in PickBestAlgorithmNoCacheCuda()
626 profile_results, std::back_inserter(filtered_results), in PickBestAlgorithmNoCacheCuda()
696 std::vector<AutotuneResult> profile_results; in PickBestAlgorithmNoCacheRocm() local
700 profile_results.emplace_back(); in PickBestAlgorithmNoCacheRocm()
701 auto& result = profile_results.back(); in PickBestAlgorithmNoCacheRocm()
739 profile_results.emplace_back(); in PickBestAlgorithmNoCacheRocm()
[all …]