Skip to content

Conversation

kavin-sai-krishna
Copy link
Contributor

This resolves #17890 .

  1. Updated kernel layout to iow/iohw as per PyTorch documentation of ConvTranspose1d and ConvTranspose2d
  2. Added support for the output_padding parameter.

@kavin-sai-krishna kavin-sai-krishna changed the title [Bugfix][Pytorch][Frontend] Bugfix of conv_transpose1d and conv_transpose2d [Bugfix][Relax][Pytorch] Bugfix of conv_transpose1d and conv_transpose2d May 13, 2025
@kavin-sai-krishna
Copy link
Contributor Author

@tvm-bot rerun

@tlopex tlopex merged commit 21a00f5 into apache:main May 14, 2025
10 checks passed
ShiboXing pushed a commit to ShiboXing/tvm that referenced this pull request Aug 10, 2025
…e2d (apache#17968)

* Bugfix conv_transpose1d and conv_transpose2d

* fix lint issue

* Update tests to reflect changes

* lint fix
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[Bug] CodeGenVM cannot handle this intrinsic now: Op(relax.nn.conv2d_transpose)
2 participants