pytorch/torch/ao/quantization
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
..
_pt2e [PT2E][Quant] Refactor quantizer and qnnpack qantizer code to support dqlinear config (#99399) 2023-05-03 03:23:32 +00:00
backend_config [quant][pt2e] remove dropout from fx quant 2023-04-27 11:22:41 -07:00
experimental AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00
fx [reland][quant][pt2e][refactor] Cleanup the logic for deciding whether to insert observer/fq or not (#99220) (#99767) 2023-04-25 16:53:02 +00:00
__init__.py [ao] making _is_activation_post_process private with BC (#90554) 2022-12-16 08:09:33 +00:00
_correct_bias.py [BE] [2/3] Rewrite super() calls in functorch and torch (#94588) 2023-02-10 21:16:33 +00:00
_equalize.py Fix typos under torch/ao directory (#97679) 2023-04-10 22:25:15 +00:00
_learnable_fake_quantize.py Fix disbale typos (#95322) 2023-02-23 02:08:45 +00:00
_quantize_pt2e.py [PT2E][Quant] Refactor quantizer and qnnpack qantizer code to support dqlinear config (#99399) 2023-05-03 03:23:32 +00:00
fake_quantize.py [BE] [2/3] Rewrite super() calls in functorch and torch (#94588) 2023-02-10 21:16:33 +00:00
fuse_modules.py Fix typos under torch/ao directory (#97679) 2023-04-10 22:25:15 +00:00
fuser_method_mappings.py Fix typos under torch/ao directory (#97679) 2023-04-10 22:25:15 +00:00
observer.py [reland][quant][pt2e][refactor] Cleanup the logic for deciding whether to insert observer/fq or not (#99220) (#99767) 2023-04-25 16:53:02 +00:00
pattern.md
qconfig_mapping.py [quant][be] Easier way to override default in QConfigMapping (#99888) 2023-04-26 18:14:01 +00:00
qconfig.py [BE] Merge isinstance calls together (#94419) 2023-02-09 00:47:26 +00:00
quant_type.py [ao] quant_type.py fixing public v private (#87519) 2022-11-15 15:42:31 +00:00
quantization_mappings.py [BE] Apply almost all remaining flake8-comprehension checks (#94676) 2023-02-12 01:01:25 +00:00
quantize_fx.py Fix typos under torch/ao directory (#97679) 2023-04-10 22:25:15 +00:00
quantize_jit.py Fix typos under torch/ao directory (#97679) 2023-04-10 22:25:15 +00:00
quantize.py Fix typos under torch/ao directory (#97679) 2023-04-10 22:25:15 +00:00
stubs.py [BE] [2/3] Rewrite super() calls in functorch and torch (#94588) 2023-02-10 21:16:33 +00:00
utils.py Return zero_point from determine_qparams as a int64 (#98746) 2023-04-11 19:01:05 +00:00