pytorch/torch/quantization
Haixin Liu cc0f1b22a2 [PyTorch Numeric Suite] Add module output comparison (#36701)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36701

Add module output comparison API.
ghstack-source-id: 103368194

Test Plan: buck test mode/dev caffe2/test:quantization -- 'test_compare_model_outputs'

Differential Revision: D21053197

fbshipit-source-id: cabcafbeeac1b604db069833a0f17ebce506ba65
2020-05-03 00:04:35 -07:00
..
__init__.py Ignore F401 in all __init__.py without putting noqa (#25823) 2019-10-23 15:28:13 -07:00
_numeric_suite.py [PyTorch Numeric Suite] Add module output comparison (#36701) 2020-05-03 00:04:35 -07:00
_quantize_script.py [quant][graph] Add check for qconfig_dict key (#37014) 2020-04-28 21:35:17 -07:00
default_mappings.py [PyTorch Numeric Suite] Add module output comparison (#36701) 2020-05-03 00:04:35 -07:00
fake_quantize.py Per channel quantization performance improvement (#33772) 2020-02-26 10:19:25 -08:00
fuse_modules.py [quant] Enable fusion for conv modules with bias (#36173) 2020-04-08 15:53:32 -07:00
observer.py Ensure that histogram observers have zero-point of zero for post ReLU activations (#37107) 2020-04-24 20:57:34 -07:00
qconfig.py [quant][graph] Add a new observer type for dynamic quantization (#35455) 2020-03-26 17:38:21 -07:00
quantize.py [PyTorch Numeric Suite] Add module output comparison (#36701) 2020-05-03 00:04:35 -07:00
stubs.py Factored out the default mappings 2019-10-03 11:52:21 -07:00