pytorch/torch/quantization
Jerry Zhang cb6b65699f [quant][graphmode][fx] Add support for packed params in state_dict (#51639)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/51639

Test Plan: Imported from OSS

Reviewed By: z-a-f

Differential Revision: D26228185

fbshipit-source-id: 6cf8b4106fec9c6900521a2afe0de6f3d29cc896
2021-02-26 15:13:50 -08:00
..
fx [quant][graphmode][fx] Add support for packed params in state_dict (#51639) 2021-02-26 15:13:50 -08:00
ns ns for fx: remove ".stats" suffix (#52799) 2021-02-25 20:45:53 -08:00
__init__.py [quant][fix] Fix quant type classification for float_qparam qconfig (#48069) 2020-11-18 18:22:08 -08:00
_correct_bias.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
_equalize.py Fix type annotation errors in torch.functional (#43446) 2020-08-26 08:27:59 -07:00
_learnable_fake_quantize.py mem-efficient learnable fake quantization (#49315) 2021-02-03 18:57:47 -08:00
_numeric_suite_fx.py compare_model_outputs_fx API implementation (#49266) 2021-02-02 10:43:25 -08:00
_numeric_suite.py ns_eager: rename Logger I/O var names to logger_cls (#51359) 2021-02-09 22:30:44 -08: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 quantization: Linear + BatchNorm1d fusion (#50748) 2021-01-20 12:59:02 -08:00
fuser_method_mappings.py quantization: Linear + BatchNorm1d fusion (#50748) 2021-01-20 12:59:02 -08:00
observer.py update HistogramObserver to be scriptable (#51081) 2021-01-27 07:27:03 -08:00
qconfig.py quant: fix conv transpose with qconfig == None (#52844) 2021-02-25 11:52:30 -08: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 [quant] Factoring out the list of no_observers (#50459) 2021-02-17 12:38:30 -08:00
quantize_fx.py [quant][graphmode][fx] Add support for packed params in state_dict (#51639) 2021-02-26 15:13:50 -08:00
quantize_jit.py [TorchScript] Support user defined classes as constants (#5062) 2020-11-16 20:52:02 -08:00
quantize.py [quant] Factoring out the list of no_observers (#50459) 2021-02-17 12:38:30 -08:00
stubs.py type check for torch.quantization.stubs (#46475) 2020-10-16 15:34:23 -07:00
utils.py [quant][refactor] Minor refactor of some typos (#50304) 2021-01-12 15:23:13 -08:00