Searched refs:use_cudnn (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 | depthwise_conv_op.cc | 375 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute() local
384 << ", Use cuDNN: " << use_cudnn; in Compute()
386 if (use_cudnn) { in Compute()
|
| D | depthwise_conv_grad_op.cc | 604 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute() local
613 << ", Use cuDNN: " << use_cudnn; in Compute()
615 if (use_cudnn) { in Compute()
1086 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute() local
1095 << ", Use cuDNN: " << use_cudnn; in Compute()
1097 if (use_cudnn) { in Compute()
|
| D | conv_ops_fused_impl.h | 91 void operator()(OpKernelContext* context, bool use_cudnn,
312 void operator()(OpKernelContext* context, bool use_cudnn,
555 void operator()(OpKernelContext* context, bool use_cudnn,
570 context, use_cudnn,
|
| 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
590 if (!use_cudnn) { 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
594 if (!use_cudnn) { 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
733 if (!use_cudnn) { in operator ()()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 969 "util/use_cudnn.h",
|