pytorch/torch/quantization
Rong Rong d1745c36dc fix type check for torch.quantization._numeric_suite (#46330)
Summary:
fix https://github.com/pytorch/pytorch/issues/42977

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

Reviewed By: malfet

Differential Revision: D24320449

Pulled By: walterddr

fbshipit-source-id: f892b5c83cb932aee53245d6c825568b3e05f3c6
2020-10-15 20:45:07 -07:00
..
fx [fx] make sure args/kwargs are immutable (#46325) 2020-10-14 15:51:43 -07:00
__init__.py Revert D24290811: [quant][eagermode] Move custom_module registration to prepare/convert_custom_config_dict 2020-10-14 16:42:55 -07: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.py fix type check for torch.quantization._numeric_suite (#46330) 2020-10-15 20:45:07 -07:00
custom_module_class_mappings.py Revert D24290811: [quant][eagermode] Move custom_module registration to prepare/convert_custom_config_dict 2020-10-14 16:42:55 -07:00
fake_quantize.py Enable type check for torch.quantization.fake_quantize (#45701) 2020-10-02 09:27:34 -07:00
fuse_modules.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
fuser_method_mappings.py [quant][eagermode][refactor] Add set/get method for quantization and fusion mappings (#43990) 2020-09-10 21:29:39 -07:00
observer.py [quant] Support 4-bit embedding_bag operators using the dtype quint4x2 (#45752) 2020-10-06 21:11:49 -07:00
qconfig.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
quant_type.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
quantization_mappings.py [quant] Refactoring the mappings files (#44847) 2020-10-14 13:15:34 -07:00
quantize_fx.py [quant][graphmode][fx] Move custom_module_class config to prepare/convert_custom_config_dict (#46251) 2020-10-14 10:43:48 -07:00
quantize_jit.py quantizaton: add API usage logging (#46095) 2020-10-09 16:51:27 -07:00
quantize.py Revert D24290811: [quant][eagermode] Move custom_module registration to prepare/convert_custom_config_dict 2020-10-14 16:42:55 -07:00
stubs.py Factored out the default mappings 2019-10-03 11:52:21 -07:00