OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
7f6a1b2bd5
pytorch/torch/quantization
History
Jerry Zhang 7f6a1b2bd5 [quant][fx][graphmode][api] Change API for custom module (#45920) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45920

See docs for new way of defining custom modules

Test Plan: Imported from OSS

Reviewed By: vkuzo

Differential Revision: D24145856

fbshipit-source-id: 488673fba503e39e8e303ed5a776fe36899ea4e3
2020-10-12 23:42:27 -07:00
..
fx [quant][fx][graphmode] Run symbolic_trace in quantization (#45919) 2020-10-08 17:26:03 -07:00
__init__.py [quant][eagermode] Custom module support (#44835) 2020-09-23 15:39:40 -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 Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
custom_module_class_mappings.py [quant][graphmode][fx] Custom module support (#44766) 2020-09-22 17:11:46 -07:00
default_mappings.py [quant] ConvTranspose1d / ConvTranspose2d (#40371) 2020-09-14 14:25:06 -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][graphmode][fx][eagermode] Add leaky relu support in quantization workflows (#45712) 2020-10-06 12:16:04 -07:00
quantize_fx.py [quant][fx][graphmode][api] Change API for custom module (#45920) 2020-10-12 23:42:27 -07:00
quantize_jit.py quantizaton: add API usage logging (#46095) 2020-10-09 16:51:27 -07:00
quantize.py quantizaton: add API usage logging (#46095) 2020-10-09 16:51:27 -07:00
stubs.py Factored out the default mappings 2019-10-03 11:52:21 -07:00