pytorch/torch/quantization
Vasiliy Kuznetsov 54da2dadd8 fx quant: more typehints, part 2 (#48792)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48792

Adds some more typehints throughout quantization/fx/quantize.py,
to help with readability.

Test Plan:
```
mypy torch/quantization/fx/quantize.py
```

Imported from OSS

Reviewed By: jerryzh168

Differential Revision: D25306683

fbshipit-source-id: fc38b885a2cb5bf2c6d23b6305658704c6eb7811
2020-12-03 19:28:12 -08:00
..
fx fx quant: more typehints, part 2 (#48792) 2020-12-03 19:28:12 -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 Compare Weights FX Implementation (#48056) 2020-11-20 17:17:19 -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 [quant] FakeQuantize inherit from FakeQuantizeBase (#48072) 2020-11-18 19:14:20 -08:00
fuse_modules.py [quant][eagermode] Add additional_fuser_method_mapping to config (#46355) 2020-10-24 02:18:04 -07:00
fuser_method_mappings.py [quant][refactor] factor out get_combined_dict function (#47781) 2020-11-11 21:01:31 -08:00
observer.py [quant][fix] Fix quant type classification for float_qparam qconfig (#48069) 2020-11-18 18:22:08 -08:00
qconfig.py [quant][fix] Fix quant type classification for float_qparam qconfig (#48069) 2020-11-18 18:22:08 -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 [reland][quant] Remove nn.quantized.ReLU module and nn.quantized.functional.relu (#47415) (#48038) 2020-11-17 09:52:21 -08:00
quantize_fx.py [qunat][graphmode][fx] Standalone module takes float as input and output (#48671) 2020-12-02 20:34:25 -08:00
quantize_jit.py [TorchScript] Support user defined classes as constants (#5062) 2020-11-16 20:52:02 -08:00
quantize.py [quantization] fix run_arg tiny bug (#48537) 2020-12-02 10:07:33 -08:00
stubs.py type check for torch.quantization.stubs (#46475) 2020-10-16 15:34:23 -07:00
utils.py [quant][refactor] Move some util functions from torch/quantization/fx/utils.py to torch/quantization/utils.py (#48107) 2020-11-18 22:32:19 -08:00