pytorch/torch/quantization/fx
Jerry Zhang f9446cb15a [quant][refactor] Remove register api and rename get_*_mapping to get_default_*_mapping (#46337)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46337

We plan to pass around the mappings instead of using global registration api to keep
the mappings local to the transformations user is performing

Test Plan: Imported from OSS

Reviewed By: vkuzo

Differential Revision: D24317436

fbshipit-source-id: 81569b88f05eeeaa9595447e482a12827aeb961f
2020-10-20 15:53:47 -07:00
..
__init__.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
fuse.py [quant][refactor] Remove register api and rename get_*_mapping to get_default_*_mapping (#46337) 2020-10-20 15:53:47 -07:00
fusion_patterns.py [quant][eagermode][refactor] Add set/get method for quantization and fusion mappings (#43990) 2020-09-10 21:29:39 -07:00
pattern_utils.py [quant][refactor] Remove register api and rename get_*_mapping to get_default_*_mapping (#46337) 2020-10-20 15:53:47 -07:00
quantization_patterns.py [fx] make sure args/kwargs are immutable (#46325) 2020-10-14 15:51:43 -07:00
quantize.py [quant][refactor] Remove register api and rename get_*_mapping to get_default_*_mapping (#46337) 2020-10-20 15:53:47 -07:00
standalone_module.py [quant][graphmode][fx] Support quantization for custom module (#44074) 2020-09-30 10:24:54 -07:00
utils.py [reland][quant][graphmode][fx] Merge all quantization mode (#45292) (#45672) 2020-10-01 15:47:11 -07:00