Searched refs:cudnn_use_autotune (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops.h | 176 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune,
186 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune,
197 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune,
207 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune,
|
| D | conv_ops.h | 36 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune,
47 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune,
|
| D | conv_ops_3d.cc | 52 static void launch(OpKernelContext* context, bool cudnn_use_autotune, in launch()
203 static void launch(OpKernelContext* ctx, bool cudnn_use_autotune, in launch()
437 if (cudnn_use_autotune && !AutoTuneConv3d::GetInstance()->Find( in launch()
|
| D | conv_ops_fused_impl.h | 92 bool cudnn_use_autotune, const Tensor& input,
313 bool cudnn_use_autotune, const Tensor& input,
556 bool cudnn_use_autotune, const Tensor& input_param,
794 if (cudnn_use_autotune) {
|
| D | conv_ops.cc | 124 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
579 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()() argument
848 if (cudnn_use_autotune && in operator ()()
|
| D | conv_grad_filter_ops.cc | 103 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
544 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()() argument
840 if (cudnn_use_autotune && !AutoTuneConvBwdFilter::GetInstance()->Find( in operator ()()
|
| D | conv_grad_input_ops.cc | 109 void operator()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
681 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()() argument
952 if (cudnn_use_autotune && !AutoTuneConvBwdData::GetInstance()->Find( in operator ()()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.h | 39 void launch(OpKernelContext* ctx, bool cudnn_use_autotune,
53 void launch(OpKernelContext* ctx, bool cudnn_use_autotune,
|
| D | fused_conv2d_bias_activation_op.cc | 441 launch(OpKernelContext* ctx, bool cudnn_use_autotune, in launch() argument
693 if (cudnn_use_autotune && !AutoTuneConvBiasActivation::GetInstance()->Find( in launch()
|