| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.cc | 293 use_cudnn_ = CanUseCudnn() && std::is_same<Device, GPUDevice>::value; in DepthwiseConv2dNativeOp()
375 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute()
405 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, in Compute()
446 bool use_cudnn_; member in tensorflow::DepthwiseConv2dNativeOp
|
| D | depthwise_conv_grad_op.cc | 567 use_cudnn_ = CanUseCudnn() && std::is_same<Device, GPUDevice>::value; in DepthwiseConv2dNativeBackpropInputOp()
604 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute()
634 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, in Compute()
660 bool use_cudnn_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp
1041 use_cudnn_ = CanUseCudnn() && std::is_same<Device, GPUDevice>::value; in DepthwiseConv2dNativeBackpropFilterOp()
1086 bool use_cudnn = use_cudnn_ && (in_depth == 1 || use_cudnn_grouped_conv_); in Compute()
1117 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute()
1143 bool use_cudnn_; member in tensorflow::DepthwiseConv2dNativeBackpropFilterOp
|
| D | conv_ops_fused_impl.h | 833 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_));
834 use_cudnn_ &= CanUseCudnn();
933 LaunchFusedConv2DOp<Device, T>()(context, use_cudnn_, cudnn_use_autotune_,
949 bool use_cudnn_;
|
| D | conv_ops.cc | 447 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_)); in Conv2DOp()
448 use_cudnn_ &= CanUseCudnn(); in Conv2DOp()
517 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, filter, in Compute()
525 bool use_cudnn_; member in tensorflow::Conv2DOp
|
| D | conv_grad_filter_ops.cc | 479 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_)); in Conv2DSlowBackpropFilterOp()
480 use_cudnn_ &= CanUseCudnn(); in Conv2DSlowBackpropFilterOp()
524 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute()
534 bool use_cudnn_; member in tensorflow::Conv2DSlowBackpropFilterOp
|
| D | conv_grad_input_ops.cc | 616 OP_REQUIRES_OK(context, context->GetAttr("use_cudnn_on_gpu", &use_cudnn_)); in Conv2DSlowBackpropInputOp()
617 use_cudnn_ &= CanUseCudnn(); in Conv2DSlowBackpropInputOp()
661 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, filter, in Compute()
671 bool use_cudnn_; member in tensorflow::Conv2DSlowBackpropInputOp
|