Jerry Zhang ace645a017 Add support for prototype affine quantization in pt2e flow (#141421) ... Summary: duplicated affine quantization functionality including observer (https://github.com/pytorch/ao/blob/main/torchao/quantization/observer.py) and some quant_primitive ops (7c3c51fd0d/torchao/quantization/quant_primitives.py (L26-L30)) to allow for per group quantization min max observer in pt2e flow Next: We can follow up to add moving average min max observer Test Plan: python test/test_quantization.py -k test_channel_group_quantization Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/141421 Approved by: https://github.com/cccclai		2024-12-24 04:22:18 +00:00
..
_internal	Add support for prototype affine quantization in pt2e flow (#141421)	2024-12-24 04:22:18 +00:00
__init__.py
_comparison.py
_creation.py
_utils.py