• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1from torch.nn.quantized import dynamic, functional, modules  # noqa: F403
2from torch.nn.quantized.modules import *  # noqa: F403
3from torch.nn.quantized.modules import MaxPool2d
4
5
6__all__ = [
7    "BatchNorm2d",
8    "BatchNorm3d",
9    "Conv1d",
10    "Conv2d",
11    "Conv3d",
12    "ConvTranspose1d",
13    "ConvTranspose2d",
14    "ConvTranspose3d",
15    "DeQuantize",
16    "Dropout",
17    "ELU",
18    "Embedding",
19    "EmbeddingBag",
20    "GroupNorm",
21    "Hardswish",
22    "InstanceNorm1d",
23    "InstanceNorm2d",
24    "InstanceNorm3d",
25    "LayerNorm",
26    "LeakyReLU",
27    "Linear",
28    "LSTM",
29    "MultiheadAttention",
30    "PReLU",
31    "Quantize",
32    "ReLU6",
33    "Sigmoid",
34    "Softmax",
35    # Wrapper modules
36    "FloatFunctional",
37    "FXFloatFunctional",
38    "QFunctional",
39]
40