tvm
tvm copied to clipboard
[Bug] nn.Conv3D groups not working correctly
nn.Conv3D is bugged
https://github.com/apache/tvm/blob/78a1f80bf24f1a1114f2ed7d17563d267bb38cc9/python/tvm/relax/frontend/nn/modules.py#L314-L325
line 325 should use in_channels instead of self.in_channels to make groups work. Otherwise in_channels calculated at line 317 would not be effectively used in weight parameters.