pytorch/torch/ao/ns/fx
HDCharles cee396fa07 [ao][ns] PNP demo for exposing arbitrary model transforms (#90153)
adding way to use arbitrary prepare and convert functions with PNP.

note this is a recreation of https://github.com/pytorch/pytorch/pull/89892 which was reverted due to landing not syncing between github and fbcode

python test/test_quantization.py
TestFxNumericSuiteNShadows.test_custom_functions_and_tracer

Differential Revision: [D41723892](https://our.internmc.facebook.com/intern/diff/D41723892/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/90153
Approved by: https://github.com/vkuzo
2022-12-06 04:24:54 +00:00
..
__init__.py
graph_matcher.py
graph_passes.py Revert "[ao] making _is_activation_post_process private (#87520)" 2022-11-21 16:48:26 +00:00
mappings.py [quant][ao_migration] nn.intrinsic.qat migration to ao (#86171) 2022-10-07 17:29:42 +00:00
n_shadows_utils.py [ao][ns] PNP demo for exposing arbitrary model transforms (#90153) 2022-12-06 04:24:54 +00:00
ns_types.py PNP: early FX numeric suite tool to quantize each layer N times (#80521) 2022-10-06 02:30:45 +00:00
pattern_utils.py [Quant][fx][bc-breaking] Rename fx/*patterns.py (#89872) 2022-12-01 17:37:07 +00:00
qconfig_multi_mapping.py [ao] making QConfigMapping print in a user friendly way (#89932) 2022-12-02 05:24:47 +00:00
utils.py Revert "[ao] making _is_activation_post_process private (#87520)" 2022-11-21 16:48:26 +00:00
weight_utils.py [quant][ao_migration] nn.intrinsic.qat migration to ao (#86171) 2022-10-07 17:29:42 +00:00