pytorch/torch/quantization
Vasiliy Kuznetsov c98c617b44 fx quant: clean up functions in _prepare (#48773)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48773

Makes util functions in `_prepare` have no side effects,
all dependencies are now in arguments.

Note: arg names are added in order as they appeared in function
code. It's not the most readable, but the lowest risk. This can
be cleaned up in future PRs if needed.

```
python test/test_quantization.py TestQuantizeFx
```

Test Plan: Imported from OSS

Reviewed By: jerryzh168

Differential Revision: D25295839

fbshipit-source-id: 60c687f6b64924473f969541c8703118e4f7d16e
2020-12-03 19:28:06 -08:00
..
fx fx quant: clean up functions in _prepare (#48773) 2020-12-03 19:28:06 -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