Searched refs:BestCudnnConvAlgorithm (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | gpu_utils.h | 244 Status BestCudnnConvAlgorithm(absl::Span<const AutotuneResult> results,
|
D | gpu_utils.cc | 246 Status BestCudnnConvAlgorithm(absl::Span<const AutotuneResult> results, in BestCudnnConvAlgorithm() function
|
D | conv_grad_input_ops.cc | 496 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in operator ()()
|
D | conv_ops_3d.cc | 620 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in launch()
|
D | conv_ops_fused_impl.h | 427 TF_RETURN_IF_ERROR(BestCudnnConvAlgorithm(results, algorithm_config));
|
D | conv_grad_ops_3d.cc | 1517 BestCudnnConvAlgorithm(results, &algorithm_config)); in Compute() 1993 Status s = BestCudnnConvAlgorithm(results, &algorithm_config); in Compute()
|
D | conv_grad_filter_ops.cc | 1095 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in operator ()()
|
D | conv_ops.cc | 1106 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in operator ()()
|