pytorch/torch/ao/quantization
Huamin Li 43853691bc [Quantization] add an option keep_original_weights in _lower_to_native_backend (#141049)
Differential Revision: D66153809

This diff adds an option to keep_original_weights so we can track back the original weight and bias after performing prepare_fx and convert_fx

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141049
Approved by: https://github.com/jerryzh168
2024-12-27 04:02:07 +00:00
..
backend_config Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
experimental remove allow-untyped-defs from torch/ao/quantization/experimental/APoT_tensor.py (#143601) 2024-12-20 05:26:09 +00:00
fx [Quantization] add an option keep_original_weights in _lower_to_native_backend (#141049) 2024-12-27 04:02:07 +00:00
pt2e Add support for prototype affine quantization in pt2e flow (#141421) 2024-12-24 04:22:18 +00:00
quantizer [reland] Kill capture_pre_autograd_graph API (#143426) 2024-12-18 12:07:09 +00:00
__init__.py Add support for prototype affine quantization in pt2e flow (#141421) 2024-12-24 04:22:18 +00:00
_correct_bias.py [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00
_equalize.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
_learnable_fake_quantize.py
fake_quantize.py
fuse_modules.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
fuser_method_mappings.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
observer.py Add support for prototype affine quantization in pt2e flow (#141421) 2024-12-24 04:22:18 +00:00
pattern.md
qconfig_mapping.py
qconfig.py
quant_type.py
quantization_mappings.py
quantize_fx.py [Quantization] add an option keep_original_weights in _lower_to_native_backend (#141049) 2024-12-27 04:02:07 +00:00
quantize_jit.py
quantize_pt2e.py [BC-Breaking]Remove capture_pre_autograd_graph references in quantization (#139505) 2024-12-13 22:26:22 +00:00
quantize.py Revert "Handle meta tensors in FX quantization (#142262)" 2024-12-21 20:34:09 +00:00
stubs.py
utils.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00