| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.cc | 294 cudnn_use_autotune_ = CudnnUseAutotune(); in DepthwiseConv2dNativeOp()
405 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, in Compute()
447 bool cudnn_use_autotune_; member in tensorflow::DepthwiseConv2dNativeOp
|
| D | depthwise_conv_grad_op.cc | 568 cudnn_use_autotune_ = CudnnUseAutotune(); in DepthwiseConv2dNativeBackpropInputOp()
634 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, in Compute()
661 bool cudnn_use_autotune_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp
1042 cudnn_use_autotune_ = CudnnUseAutotune(); in DepthwiseConv2dNativeBackpropFilterOp()
1117 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute()
1144 bool cudnn_use_autotune_; member in tensorflow::DepthwiseConv2dNativeBackpropFilterOp
|
| D | conv_ops_3d.cc | 114 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv3DOp()
168 LaunchConvOp<Device, T>::launch(context, cudnn_use_autotune_, input, filter, in Compute()
178 bool cudnn_use_autotune_; member in tensorflow::Conv3DOp
|
| D | conv_grad_ops_3d.cc | 1136 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv3DBackpropInputOp()
1358 if (cudnn_use_autotune_ && !AutoTuneConv3dBwdData::GetInstance()->Find( in Compute()
1461 bool cudnn_use_autotune_; member in tensorflow::Conv3DBackpropInputOp
1519 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv3DBackpropFilterOp()
1764 if (cudnn_use_autotune_ && !AutoTuneConv3dBwdFilter::GetInstance()->Find( in Compute()
1844 bool cudnn_use_autotune_; member in tensorflow::Conv3DBackpropFilterOp
|
| D | conv_ops_fused_impl.h | 835 cudnn_use_autotune_ = CudnnUseAutotune();
933 LaunchFusedConv2DOp<Device, T>()(context, use_cudnn_, cudnn_use_autotune_,
950 bool cudnn_use_autotune_;
|
| D | conv_ops.cc | 449 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv2DOp()
517 launcher_(context, use_cudnn_, cudnn_use_autotune_, input, filter, in Compute()
526 bool cudnn_use_autotune_; member in tensorflow::Conv2DOp
|
| D | conv_grad_filter_ops.cc | 481 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv2DSlowBackpropFilterOp()
524 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute()
537 bool cudnn_use_autotune_; member in tensorflow::Conv2DSlowBackpropFilterOp
|
| D | conv_grad_input_ops.cc | 618 cudnn_use_autotune_ = CudnnUseAutotune(); in Conv2DSlowBackpropInputOp()
661 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, filter, in Compute()
674 bool cudnn_use_autotune_; member in tensorflow::Conv2DSlowBackpropInputOp
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 151 cudnn_use_autotune_ = CudnnUseAutotune(); in FusedConv2DBiasActivationOp()
240 launcher_.launch(context, cudnn_use_autotune_, conv_input, conv_input_scale, in Compute()
254 bool cudnn_use_autotune_; member in tensorflow::FusedConv2DBiasActivationOp
|