diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-30 06:59:23 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-30 07:01:55 -0700 |
commit | a5a51ad3a1200e2e5ef46c140bab717422e41ca2 (patch) | |
tree | 7c1e5ee890228c81ef57d6c6998a04dac79a29e2 /tensorflow/stream_executor/dnn.cc | |
parent | 914796d5e9bc7b0c619b53c7eb24cfe7d6c7fb9b (diff) |
Adding a depthwise convolution kernel op (with label 'cudnn_grouped_convolution') which forwards to cuDNN grouped convolutions.
PiperOrigin-RevId: 194780352
Diffstat (limited to 'tensorflow/stream_executor/dnn.cc')
-rw-r--r-- | tensorflow/stream_executor/dnn.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dnn.cc b/tensorflow/stream_executor/dnn.cc index 031c82d3f4..eed93efc8d 100644 --- a/tensorflow/stream_executor/dnn.cc +++ b/tensorflow/stream_executor/dnn.cc @@ -434,6 +434,7 @@ ConvolutionDescriptor::ConvolutionDescriptor(int ndims) filter_strides_(ndims, 1), dilation_rates_(ndims, 1), pad_alignment_(PadAlignment::kDefault), + group_count_(1), ndims_(ndims) {} ConvolutionDescriptor::ConvolutionDescriptor() |