feat(aten::conv_transpose): Add support for dilated and group

deconvolution Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Jul 14, 2020 · 48b950a · 48b950a
1 parent 842a567
commit 48b950a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/core/conversion/converters/impl/conv_deconv.cpp b/core/conversion/converters/impl/conv_deconv.cpp
@@ -45,6 +45,9 @@ auto conv_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns()
 
                 deconv->setStrideNd(stride);
                 deconv->setPaddingNd(padding);
+                deconv->setDilationNd(dilation);
+                deconv->setNbGroups(groups);
+
                 new_layer = deconv;
             } else {
                 nvinfer1::IConvolutionLayer* conv;