Lyken17 commented on pull request #9465: URL: https://github.com/apache/tvm/pull/9465#issuecomment-964702295
Seems we are in consensus now, 1. `Groups` support should be added for `conv2d_transpose`. 2. The weight layout of `conv2d_transpose` should be `IOHW`, a.k.a, (input_channel, output_channel // groups, kernel_height, kernel_width). Second point involves a series of changes on `topi.nn`, `topi.strategy` and `src/relay/op/convolution.h` and only update `conv2d_transpose` will make it inconsistent with `conv1d_tranpose` and `conv3d_transpose`. Since there is ticket blocking v0.8.0 release https://github.com/apache/tvm/issues/8976, I am concerned that I might not be able to update all in one week. Let me explore whether it is possible to keep `OIHW` for now while supporting gradients and groups > 1. -- 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. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
