Go
pytorch
25aad029
- Only extract bias out of linear layer if input ndims >= 4 (#1679)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
2 years ago
Only extract bias out of linear layer if input ndims >= 4 (#1679) restrict decomposition of linear for input.rank() >=4 updating python tests Co-authored-by: jjsjann123 <jiej@nvidia.com>
Author
csarofeen
Parents
04c5ac65
Loading