Searched refs:BestCudnnConvAlgorithm (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.h | 181 Status BestCudnnConvAlgorithm(absl::Span<const AutotuneResult> results,
|
| D | gpu_utils.cc | 110 Status BestCudnnConvAlgorithm(absl::Span<const AutotuneResult> results, in BestCudnnConvAlgorithm() function
|
| D | conv_ops_3d.cc | 480 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in launch()
|
| D | conv_ops_fused_impl.h | 548 TF_RETURN_IF_ERROR(BestCudnnConvAlgorithm(results, algorithm_config));
|
| D | conv_ops.cc | 890 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in operator ()()
|
| D | conv_grad_filter_ops.cc | 879 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in operator ()()
|
| D | conv_grad_input_ops.cc | 990 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in operator ()()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 343 Status BestCudnnConvAlgorithm(absl::Span<const AutotuneResult> results, in BestCudnnConvAlgorithm() function
748 ctx, internal::BestCudnnConvAlgorithm(results, &algorithm_config)); in launch()
|