diff options
author | 2017-11-29 16:38:32 -0800 | |
---|---|---|
committer | 2017-11-29 16:42:20 -0800 | |
commit | cb4ef362e4a18b3c42a2c90bdad8754d5ead4caf (patch) | |
tree | cca762531469418ade8c8d5bdf2a9ba6719ec111 /tensorflow/core/kernels/depthwise_conv_op.cc | |
parent | 4ada275eed7472ae32c67a1ec0b9b1dc8d80d1f0 (diff) |
Add native dilated support for conv2d and its gradients in cudnn v>=6.
PiperOrigin-RevId: 177382431
Diffstat (limited to 'tensorflow/core/kernels/depthwise_conv_op.cc')
-rw-r--r-- | tensorflow/core/kernels/depthwise_conv_op.cc | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/depthwise_conv_op.cc b/tensorflow/core/kernels/depthwise_conv_op.cc index 7c43dcb670..02da64ce98 100644 --- a/tensorflow/core/kernels/depthwise_conv_op.cc +++ b/tensorflow/core/kernels/depthwise_conv_op.cc @@ -373,8 +373,11 @@ class DepthwiseConv2dNativeOp : public BinaryOp<T> { // If in_depth==1, this operation is just a standard convolution, so // invoke that op. if (std::is_same<T, float>::value && in_depth == 1) { + // TODO(yangzihao): Send in arbitrary dilation rates after the dilated + // conv is supported. launcher_(context, use_cudnn_, cudnn_use_autotune_, input, filter, - stride_, stride_, padding_, output, data_format_); + /*row_dilation=*/1, /*col_dilation=*/1, stride_, stride_, + padding_, output, data_format_); return; } |