__init__.py 366 B

123456789101112131415161718
  1. # flake8: noqa: F401
  2. r"""QAT Dynamic Modules
  3. This package is in the process of being deprecated.
  4. Please, use `torch.ao.nn.qat.dynamic` instead.
  5. """
  6. from . import dynamic # noqa: F403
  7. from . import modules # noqa: F403
  8. from .modules import * # noqa: F403
  9. __all__ = [
  10. "Linear",
  11. "Conv1d",
  12. "Conv2d",
  13. "Conv3d",
  14. "Embedding",
  15. "EmbeddingBag",
  16. ]