| /external/tensorflow/tensorflow/core/kernels/ |
| depthwise_conv_op.cc | 288 use_cudnn_ = CanUseCudnn();
378 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, filter,
415 bool use_cudnn_; member in class:tensorflow::DepthwiseConv2dNativeOp
|
| conv_ops.cc | 266 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_));
267 use_cudnn_ &= CanUseCudnn();
422 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, filter,
430 bool use_cudnn_; member in class:tensorflow::Conv2DOp
[all...] |
| conv_grad_filter_ops.cc | 572 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_));
573 use_cudnn_ &= CanUseCudnn();
613 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input,
622 bool use_cudnn_; member in class:tensorflow::Conv2DSlowBackpropFilterOp
[all...] |
| conv_grad_input_ops.cc | 647 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_));
648 use_cudnn_ &= CanUseCudnn();
682 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, filter,
691 bool use_cudnn_; member in class:tensorflow::Conv2DSlowBackpropInputOp
[all...] |
