Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc260 auto selected_result = filtered_results.begin(); in BestCudnnConvAlgorithm() local
268 selected_result = absl::c_min_element(filtered_results, compare_run_times); in BestCudnnConvAlgorithm()
273 algo->set_algorithm({selected_result->conv().algorithm(), in BestCudnnConvAlgorithm()
274 selected_result->conv().tensor_ops_enabled()}); in BestCudnnConvAlgorithm()
275 algo->set_scratch_size(selected_result->scratch_bytes()); in BestCudnnConvAlgorithm()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc638 auto selected_result = filtered_results.begin(); in PickBestAlgorithmNoCacheCuda() local
641 selected_result = absl::c_min_element( in PickBestAlgorithmNoCacheCuda()
649 return *selected_result; in PickBestAlgorithmNoCacheCuda()