vinx13 commented on a change in pull request #4476: Implement 1d deconvolution
URL: https://github.com/apache/incubator-tvm/pull/4476#discussion_r357821616
##########
File path: python/tvm/relay/op/nn/_nn.py
##########
@@ -348,6 +348,38 @@ def legalize_conv2d_transpose(attrs, inputs, types):
reg.register_pattern("nn.conv2d_transpose", OpPattern.OUT_ELEMWISE_FUSABLE)
+# conv1d_transpose
[email protected]_compute("nn.conv1d_transpose")
+def compute_conv1d_transpose(attrs, inputs, out_dtype, target):
+ """Compute definition of conv1d_transpose"""
+ padding = get_const_tuple(attrs.padding)
+ strides = get_const_tuple(attrs.strides)
+ dilation = get_const_tuple(attrs.dilation)
+ groups = attrs.groups
+ layout = attrs.data_layout
+ out_dtype = attrs.out_dtype
+ out_dtype = (inputs[0].dtype if out_dtype in ("same", "")
+ else out_dtype)
+ assert layout == "NCW", "conv1d_transpose ncw only supported"
+ assert dilation == (1,), "conv1d_transpose dilation is not supported"
+ assert groups == 1, "conv1d_transpose groups == 1 only supported"
+ out = topi.nn.conv1d_transpose_ncw(
+ inputs[0], inputs[1], strides, padding, out_dtype)
+ output_padding = get_const_tuple(attrs.output_padding)
+ out = topi.nn.pad(out,
+ [0, 0, 0], [0, 0, output_padding[0]])
+ return [out]
+
+
[email protected]_schedule("nn.conv1d_transpose")
+def schedule_conv1d_transpose(attrs, outs, target):
+ """Schedule definition of conv1d_transpose"""
+ with target:
+ return topi.generic.schedule_conv1d_transpose_ncw(outs)
+ raise ValueError("No compatible schedule")
Review comment:
remove this line
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
