mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
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
25 lines
907 B
Python
25 lines
907 B
Python
# flake8: noqa: F401
|
|
r"""
|
|
This file is in the process of migration to `torch/ao/quantization`, and
|
|
is kept here for compatibility while the migration process is ongoing.
|
|
If you are adding a new entry/functionality, please, add it to the
|
|
`torch/ao/quantization/fuse_modules.py`, while adding an import statement
|
|
here.
|
|
"""
|
|
|
|
from torch.ao.quantization.fuse_modules import fuse_modules
|
|
from torch.ao.quantization.fuse_modules import fuse_known_modules
|
|
from torch.ao.quantization.fuse_modules import get_fuser_method
|
|
|
|
# for backward compatiblity
|
|
from torch.quantization.fuser_method_mappings import fuse_conv_bn
|
|
from torch.quantization.fuser_method_mappings import fuse_conv_bn_relu
|
|
|
|
# TODO: These functions are not used outside the `fuse_modules.py`
|
|
# Keeping here for now, need to remove them later.
|
|
from torch.ao.quantization.fuse_modules import (
|
|
_fuse_modules,
|
|
_get_module,
|
|
_set_module,
|
|
)
|