pytorch/torch/quantization
yanli924 ada916675f update HistogramObserver to be scriptable (#51081)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51081

Pull Request resolved: https://github.com/pytorch/pytorch/pull/51001

fix tests in TestQuantizeJitOps

Test Plan:
Imported from OSS
python test/test_quantization.py

Reviewed By: raghuramank100

Differential Revision: D26038759

Pulled By: lyoka

fbshipit-source-id: 0977ba7b8b26a9f654f20f5c698a7a20ec078c35
2021-01-27 07:27:03 -08:00
..
fx [quant][graphmode][fx] Add support for quantizing functional linear + {functional relu/module relu} (#50975) 2021-01-25 12:49:58 -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 Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
_numeric_suite_fx.py Clean up some type annotations in caffe2/torch/quantization (#49942) 2020-12-29 15:43:50 -08:00
_numeric_suite.py T78750158 Support varying size input in numeric suite at 10/30/2020, 3:55:01 PM (#47391) 2020-11-18 23:57:41 -08:00
fake_quantize.py fake_quant: fix device affinity and buffer resizing for state_dict (#50868) 2021-01-25 13:50:28 -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: nice error message on convtranspose with per-channel weight (#49899) 2021-01-05 09:38:57 -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][bug] Fixing the mapping getter to return a copy (#50297) 2021-01-12 15:19:39 -08:00
quantize_fx.py [quant][graphmode][fx] Support preserved_attributes in prepare_fx (#50306) 2021-01-11 12:10:02 -08:00
quantize_jit.py [TorchScript] Support user defined classes as constants (#5062) 2020-11-16 20:52:02 -08:00
quantize.py eager quant: fix error with removing forward hooks (#49813) 2021-01-05 11:00:20 -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