Jerry Zhang 74454bdb46 [quant][fx] Move backend_config folder to torch.ao.quantization ...

Summary:
Following https://github.com/pytorch/rfcs/blob/master/RFC-0019-Extending-PyTorch-Quantization-to-Custom-Backends.md we implemented
the backend configuration for fbgemm/qnnpack backend, currently it was under fx folder, but we'd like to use this for all different
workflows, including eager, fx graph and define by run quantization, this PR moves it to torch.ao.quantization namespace so that
it can be shared by different workflows
Also moves some utility functions specific to fx to fx/backend_config_utils.py and some files are kept in fx folder (quantize_handler.py and fuse_handler.py)

Test Plan:
python test/teset_quantization.py TestQuantizeFx
python test/teset_quantization.py TestQuantizeFxOps
python test/teset_quantization.py TestQuantizeFxModels
python test/test_quantization.py TestAOMigrationQuantization
python test/test_quantization.py TestAOMigrationQuantizationFx

Reviewers:

Subscribers:

Tasks:

Tags:

Pull Request resolved: https://github.com/pytorch/pytorch/pull/75823

Approved by: https://github.com/vkuzo

2022-04-19 15:38:57 +00:00

..

__init__.py

[quant] Rename _convert_do_not_use.py to convert.py (#74322)

2022-03-17 18:57:08 +00:00

_equalize.py

[quant] AO migration of the torch/quantization/quantize_fx.py and torch/quantization/fx/* (#65033)

2021-09-22 09:29:15 -07:00

convert.py

[quant] Rename _convert_do_not_use.py to convert.py (#74322)

2022-03-17 18:57:08 +00:00

fuse.py

[quant][fx] Remove Fuser class in fusion implementation (#73470)

2022-03-01 09:29:21 +00:00

fusion_patterns.py

[reland][quant][graphmode][fx] Enable fuse handler for sequence of 3 ops (#70006)

2021-12-16 15:04:16 -08:00

graph_module.py

[quant] AO migration of the torch/quantization/quantize_fx.py and torch/quantization/fx/* (#65033)

2021-09-22 09:29:15 -07:00

match_utils.py

[quant] AO migration of the torch/quantization/quantize_fx.py and torch/quantization/fx/* (#65033)

2021-09-22 09:29:15 -07:00

pattern_utils.py

[quant][fx] Move backend_config folder to torch.ao.quantization

2022-04-19 15:38:57 +00:00

prepare.py

[quant] AO migration of the torch/quantization/quantize_fx.py and torch/quantization/fx/* (#65033)

2021-09-22 09:29:15 -07:00

quantization_patterns.py

[quant][fx] Move backend_config folder to torch.ao.quantization

2022-04-19 15:38:57 +00:00

quantization_types.py

[quant][fx] Move backend_config folder to torch.ao.quantization

2022-04-19 15:38:57 +00:00

utils.py

[ao] Fixing obs insertion through dtype propagation (#73274)

2022-03-16 01:41:17 +00:00