pytorch/torch/ao
asl3 4680047001 Modify LinearAPoT matrix multiplication bitshift to support all k (#82409)
### Summary
This PR modifies the bitshift implementation of matrix multiplication for LinearAPoT in `bitshift_mul` to support all input values of k. It also fixes the row/col dimension assignment for the `mat_mul `method

### Test Plan
Run unit tests with: `python test/quantization/core/experimental/test_linear.py`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82409
Approved by: https://github.com/dzdang
2022-07-28 20:40:26 +00:00
..
nn [PyTorch Edge] Add qnnpack bcsr matrix unpacking and use unpacking in Linear module (#80475) 2022-07-07 15:32:21 +00:00
ns Add prelu op and module for quantized CPU backend (#73491) 2022-07-20 07:48:15 +00:00
quantization Modify LinearAPoT matrix multiplication bitshift to support all k (#82409) 2022-07-28 20:40:26 +00:00
sparsity [ao][sparsity] README for base data scheduler class (#82131) 2022-07-27 20:44:38 +00:00
__init__.py [AO] Clear the contents of the torch/ao/__init__.py (#69415) 2021-12-09 01:21:30 -08:00