1# flake8: noqa: F401 2r""" 3This file is in the process of migration to `torch/ao/quantization`, and 4is kept here for compatibility while the migration process is ongoing. 5If you are adding a new entry/functionality, please, add it to the 6`torch/ao/quantization/fuse_modules.py`, while adding an import statement 7here. 8""" 9 10# TODO: These functions are not used outside the `fuse_modules.py` 11# Keeping here for now, need to remove them later. 12from torch.ao.quantization.fuse_modules import ( 13 _fuse_modules, 14 _get_module, 15 _set_module, 16 fuse_known_modules, 17 fuse_modules, 18 get_fuser_method, 19) 20 21# for backward compatiblity 22from torch.ao.quantization.fuser_method_mappings import fuse_conv_bn, fuse_conv_bn_relu 23