Home
last modified time | relevance | path

Searched defs:use_cudnn (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.cc375 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute() local
Ddepthwise_conv_grad_op.cc604 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute() local
1086 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute() local
Dconv_grad_filter_ops.cc544 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
Dconv_grad_input_ops.cc681 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
Dconv_ops.cc579 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()