| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.cc | 301 cudnn_use_autotune_ = CudnnUseAutotune(); in DepthwiseConv2dNativeOp()
444 launcher_(context, /*use_cudnn=*/true, cudnn_use_autotune_, input, in Compute()
486 bool cudnn_use_autotune_; member in tensorflow::DepthwiseConv2dNativeOp
|
| D | depthwise_conv_grad_op.cc | 582 cudnn_use_autotune_ = CudnnUseAutotune(); in DepthwiseConv2dNativeBackpropInputOp()
675 launcher_(context, /*use_cudnn=*/true, cudnn_use_autotune_, out_backprop, in Compute()
702 bool cudnn_use_autotune_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp
1085 cudnn_use_autotune_ = CudnnUseAutotune(); in DepthwiseConv2dNativeBackpropFilterOp()
1173 launcher_(context, /*use_cudnn=*/true, cudnn_use_autotune_, out_backprop, in Compute()
1233 bool cudnn_use_autotune_; member in tensorflow::DepthwiseConv2dNativeBackpropFilterOp
|
| D | conv_grad_ops_3d.cc | 1149 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv3DBackpropInputOp()
1422 cudnn_use_autotune_ = true; in Compute()
1425 if (cudnn_use_autotune_ && !AutoTuneConv3dBwdData::GetInstance()->Find( in Compute()
1572 bool cudnn_use_autotune_; member in tensorflow::Conv3DBackpropInputOp
1630 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv3DBackpropFilterOp()
1919 cudnn_use_autotune_ = true; in Compute()
1922 if (cudnn_use_autotune_ && !AutoTuneConv3dBwdFilter::GetInstance()->Find( in Compute()
2038 bool cudnn_use_autotune_; member in tensorflow::Conv3DBackpropFilterOp
|
| D | conv_ops_3d.cc | 120 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv3DOp()
177 LaunchConvOp<Device, T>::launch(context, cudnn_use_autotune_, input, filter, in Compute()
187 bool cudnn_use_autotune_; member in tensorflow::Conv3DOp
|
| D | conv_grad_input_ops.h | 401 cudnn_use_autotune_ = CudnnUseAutotune();
454 launch(context, use_cudnn_, cudnn_use_autotune_, out_backprop, filter,
467 bool cudnn_use_autotune_ = false;
|
| D | conv_ops_fused_impl.h | 724 cudnn_use_autotune_ = CudnnUseAutotune();
796 LaunchFusedConv2DOp<Device, T>()(context, use_cudnn_, cudnn_use_autotune_,
805 bool cudnn_use_autotune_;
|
| D | conv_grad_filter_ops.cc | 304 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv2DBackpropFilterOp()
361 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute()
374 bool cudnn_use_autotune_; member in tensorflow::Conv2DBackpropFilterOp
|
| D | conv_ops.cc | 515 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv2DOp()
583 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, filter, in Compute()
592 bool cudnn_use_autotune_; member in tensorflow::Conv2DOp
|