Home
last modified time | relevance | path

Searched refs:best_result_no_scratch (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc125 const AutotuneResult* best_result_no_scratch = std::min_element( in BestCudnnConvAlgorithm() local
141 if (best_result_no_scratch != results.end() && in BestCudnnConvAlgorithm()
142 best_result_no_scratch->has_success() && in BestCudnnConvAlgorithm()
143 best_result_no_scratch->success().scratch_bytes() == 0) { in BestCudnnConvAlgorithm()
145 {best_result_no_scratch->conv().algorithm(), in BestCudnnConvAlgorithm()
146 best_result_no_scratch->conv().tensor_ops_enabled()}); in BestCudnnConvAlgorithm()
Dconv_grad_ops_3d.cc1366 ProfileResult best_result_no_scratch; in Compute() local
1388 best_result_no_scratch.elapsed_time_in_ms()) { in Compute()
1389 best_result_no_scratch = profile_result; in Compute()
1395 best_result.is_valid() || best_result_no_scratch.is_valid(), in Compute()
1400 if (best_result_no_scratch.is_valid()) { in Compute()
1402 best_result_no_scratch.algorithm()); in Compute()
1772 ProfileResult best_result_no_scratch; in Compute() local
1795 best_result_no_scratch.elapsed_time_in_ms()) { in Compute()
1796 best_result_no_scratch = profile_result; in Compute()
1802 best_result.is_valid() || best_result_no_scratch.is_valid(), in Compute()
[all …]
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc358 const AutotuneResult* best_result_no_scratch = std::min_element( in BestCudnnConvAlgorithm() local
374 if (best_result_no_scratch != results.end() && in BestCudnnConvAlgorithm()
375 best_result_no_scratch->has_success() && in BestCudnnConvAlgorithm()
376 best_result_no_scratch->success().scratch_bytes() == 0) { in BestCudnnConvAlgorithm()
378 {best_result_no_scratch->conv().algorithm(), in BestCudnnConvAlgorithm()
379 best_result_no_scratch->conv().tensor_ops_enabled()}); in BestCudnnConvAlgorithm()