• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# flake8: noqa: F401
2r"""QAT Modules.
3
4This package is in the process of being deprecated.
5Please, use `torch.ao.nn.qat.modules` instead.
6"""
7from torch.ao.nn.qat.modules.conv import Conv1d, Conv2d, Conv3d
8from torch.ao.nn.qat.modules.embedding_ops import Embedding, EmbeddingBag
9from torch.ao.nn.qat.modules.linear import Linear
10from torch.nn.qat.modules import conv, embedding_ops, linear
11
12
13__all__ = [
14    "Linear",
15    "Conv1d",
16    "Conv2d",
17    "Conv3d",
18    "Embedding",
19    "EmbeddingBag",
20]
21