[quant] AO migration of the `fuse_modules.py` (phase 1) (#64913)
authorZafar Takhirov <zaf@fb.com>
Thu, 16 Sep 2021 00:24:09 +0000 (17:24 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 16 Sep 2021 00:28:47 +0000 (17:28 -0700)
commita42996f16e165de2c7f301da74a4de7eb3c0f452
tree0b8a7e38523c3fd926eeeb9a86da7e348406d8a0
parent7e9c599784a3c4496161cc96dbc9e48afa613d83
[quant] AO migration of the `fuse_modules.py` (phase 1) (#64913)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64913

AO Team is migrating the existing torch.quantization into torch.ao.quantization. We are doing it one file at a time to make sure that the internal callsites are updated properly.
This migrates the fuse_module.py from torch.quantization to torch.ao.quantization.
At this point both locations will be supported. Eventually the torch.quantization will be deprecated.

Test Plan: `buck test mode/dev //caffe2/test:quantization`

Reviewed By: vkuzo

Differential Revision: D30882819

fbshipit-source-id: 1926ad6aa49136aceb5b625dcef4bfde3a2860d4
test/quantization/ao_migration/test_quantization.py
test/quantization/core/test_workflow_module.py
test/quantization/eager/test_equalize_eager.py
test/quantization/eager/test_model_numerics.py
test/quantization/eager/test_quantize_eager_ptq.py
torch/ao/quantization/__init__.py
torch/ao/quantization/fuse_modules.py [new file with mode: 0644]
torch/quantization/fuse_modules.py