pytorch
25aad029 - Only extract bias out of linear layer if input ndims >= 4 (#1679)

Commit
3 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
Parents
Loading