pytorch/torch/quantization
Jerry Zhang f2b851a9e5 Returning axis from calculate_qparams (#29494)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29494

`calculate_qparams` of per channel quantization should return the axis, this
PR added this and also added corresponding support in graph mode

Test Plan:
python test/test_jit.py

Imported from OSS

Differential Revision: D18580905

fbshipit-source-id: f9691c1f043f8bca39f81716a4d0b10f60a65396
2019-11-20 11:06:48 -08:00
..
__init__.py Ignore F401 in all __init__.py without putting noqa (#25823) 2019-10-23 15:28:13 -07:00
_quantize_script.py Convert conv_prepack to conv2d_prepack and conv_unpack to conv2d_unpack (#29529) 2019-11-11 21:54:10 -08:00
default_mappings.py Add nn.quantized.Conv3d (#29813) 2019-11-15 04:33:40 -08:00
fake_quantize.py Returning axis from calculate_qparams (#29494) 2019-11-20 11:06:48 -08:00
fuse_modules.py Rename _intrinsic to intrinsic 2019-10-02 18:53:06 -07:00
observer.py Returning axis from calculate_qparams (#29494) 2019-11-20 11:06:48 -08:00
qconfig.py Refactoing names for consistency 2019-10-16 12:18:26 -07:00
quantize.py Do not insert observers for empty sequential modules (#28384) 2019-10-21 20:32:13 -07:00
stubs.py Factored out the default mappings 2019-10-03 11:52:21 -07:00