linear_fused.py 453 B

123456789101112131415
  1. # flake8: noqa: F401
  2. r"""Intrinsic QAT Modules
  3. This file is in the process of migration to `torch/ao/nn/intrinsic/qat`, 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. appropriate file under the `torch/ao/nn/intrinsic/qat/modules`,
  7. while adding an import statement here.
  8. """
  9. __all__ = [
  10. 'LinearBn1d',
  11. ]
  12. from torch.ao.nn.intrinsic.qat import LinearBn1d