Jerry Zhang 7523728368 [quant][graphmode][fx] Factor out run_weight_observer (#59029) ...

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/59029

Trying to remove Quantizer class and split prepare and convert code

Test Plan:
python test/test_quantization.py TestQuantizeFx
python test/test_quantization.py TestQuantizeFxOps

Imported from OSS

Reviewed By: vkuzo

Differential Revision: D28724864

fbshipit-source-id: 67ac5e7eb351970fdf46532c3c2ac6ac831bc697

2021-06-01 10:01:42 -07:00

..

fx

[quant][graphmode][fx] Factor out run_weight_observer (#59029)

2021-06-01 10:01:42 -07:00

ns

fix nn.MHA scriptability (#58727)

2021-05-26 15:29:49 -07:00

__init__.py

Un-ignore F403 in .flake8 (#55838)

2021-04-13 09:24:07 -07:00

_correct_bias.py

Remove py2 compatible future imports (#44735)

2020-09-16 12:55:57 -07:00

_equalize.py

[quant] Eager mode equalization support for ConvReLU and LinearReLU (#58792)

2021-05-24 17:25:13 -07:00

_learnable_fake_quantize.py

Add lint for unqualified type: ignore (#56290)

2021-04-21 08:07:23 -07:00

_numeric_suite_fx.py

ns for fx: move relatedness mapping to mappings file (#57171)

2021-05-05 06:29:11 -07:00

_numeric_suite.py

Add lint for unqualified type: ignore (#56290)

2021-04-21 08:07:23 -07:00

fake_quantize.py

memory efficient per-channel fq: use it everywhere, delete old version (#51265)

2021-01-28 19:42:25 -08:00

fuse_modules.py

Add lint for unqualified noqa (#56272)

2021-04-19 13:16:18 -07:00

fuser_method_mappings.py

fix docstring for fusing functions (#58638)

2021-05-24 18:27:22 -07:00

observer.py

[QAT] Fix the runtime run cannot resize variables that require grad (#57068)

2021-05-27 09:12:06 -07:00

qconfig.py

Un-ignore F403 in .flake8 (#55838)

2021-04-13 09:24:07 -07:00

quant_type.py

[quant][graphmode][fx] custom_module support static/dynamic/weight_only quant (#46786)

2020-10-27 21:41:33 -07:00

quantization_mappings.py

fix nn.MHA scriptability (#58727)

2021-05-26 15:29:49 -07:00

quantize_fx.py

[quant][fx] Validate qconfig_dict keys (#58566)

2021-05-21 15:20:05 -07:00

quantize_jit.py

Enable the quantization on XPU devices (#54857)

2021-05-20 17:02:13 -07:00

quantize.py

Back out "[quant][graphmode][fx] Separate handling Copy operator to a helper function" (#55388)

2021-04-06 14:20:36 -07:00

stubs.py

type check for torch.quantization.stubs (#46475)

2020-10-16 15:34:23 -07:00

utils.py

[quant][graphmode][fx] Fix a condition check for CopyNode (#53585)

2021-03-11 09:32:20 -08:00