fuser_method_mappings.py 511 B

123456789101112131415
  1. # flake8: noqa: F401
  2. r"""
  3. This file is in the process of migration to `torch/ao/quantization`, and
  4. is kept here for compatibility while the migration process is ongoing.
  5. If you are adding a new entry/functionality, please, add it to the
  6. `torch/ao/quantization/fuser_method_mappings.py`, while adding an import statement
  7. here.
  8. """
  9. from torch.ao.quantization.fuser_method_mappings import (
  10. fuse_conv_bn,
  11. fuse_conv_bn_relu,
  12. fuse_linear_bn,
  13. _DEFAULT_OP_LIST_TO_FUSER_METHOD,
  14. get_fuser_method,
  15. )