Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Conv2DTransposed] Fix wrong shape check
The default shape format of TVM is `N x Cx iH x iW` for input and `O x I x kH x kW` for weight, a proper shape for Conv2dTransposed should * input: (batch, in_channels, iH, iW) * weight: (out_channels, in_channels // groups, kH, kW) Thus the original checking ``` ICHECK(reporter->AssertEQ(indexdiv(dshape_nchw[1], param->groups), wshape[0])); ``` is wrong. The proper comparison dimension should be `wshape[1]` rather than `wshape[0]`. Besides, the name for debug is also not correct. All logging information are using `conv2d` rather than `conv2d_transposed`, which is confusing.
- Loading branch information