pytorch/torch/ao
Kimish Patel 771a9debbe [PT2E][Quant] Refactor quantizer and qnnpack qantizer code to support dqlinear config (#99399)
This diff introduces a few refactors:

- Move observer creation to utils.py.
- Use quantization spec to supply args to observers.
- Use annotation function registration corresponding QuantizationConfig. This
  will be later used in dynamic quantized linear.

Differential Revision: [D45073790](https://our.internmc.facebook.com/intern/diff/D45073790/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/99399
Approved by: https://github.com/jerryzh168
2023-05-03 03:23:32 +00:00
..
nn [quant][pt2e] Fix a bug in reference quantized module (decomposed mode) (#98903) 2023-04-13 21:55:45 +00:00
ns Fix flake8 lint errors reported by ruff - take 2 (#99798) 2023-04-23 23:09:51 +00:00
pruning [BE] Enable C419 rule for any all shortcircuiting (#99890) 2023-04-25 15:02:13 +00:00
quantization [PT2E][Quant] Refactor quantizer and qnnpack qantizer code to support dqlinear config (#99399) 2023-05-03 03:23:32 +00:00
__init__.py [refactor] Renaming ao.sparsity to ao.pruning (#84867) 2022-10-07 00:58:41 +00:00