Jerry Zhang 882eb0f646 [quant][graphmode][fx] Add support for dynamic quant for RNN and RNNCell (#49126) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/49126 Test Plan: python test/test_quantization.py TestQuantizeFxOps.test_rnn python test/test_quantization.py TestQuantizeFxOps.test_rnn_cell Imported from OSS Reviewed By: raghuramank100 Differential Revision: D25449047 fbshipit-source-id: 532bf9ad2839958dde8c6f2d9399fac96b2b8bd4		2020-12-10 18:11:40 -08:00
..
fx	[quant][graphmode][fx] Add support for dynamic quant for RNN and RNNCell (#49126)	2020-12-10 18:11:40 -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	[quant][graphmode][fx] Add support for dynamic quant for RNN and RNNCell (#49126)	2020-12-10 18:11:40 -08:00
quantize_fx.py	[quant][fx][graphmode] Renable torchvision test (#48602)	2020-12-04 10:13:38 -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