• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# flake8: noqa: F401
2r"""Intrinsic QAT Modules.
3
4This file is in the process of migration to `torch/ao/nn/intrinsic/qat`, and
5is kept here for compatibility while the migration process is ongoing.
6If you are adding a new entry/functionality, please, add it to the
7appropriate file under the `torch/ao/nn/intrinsic/qat/modules`,
8while adding an import statement here.
9"""
10
11from torch.ao.nn.intrinsic.qat import (
12    ConvBn1d,
13    ConvBn2d,
14    ConvBn3d,
15    ConvBnReLU1d,
16    ConvBnReLU2d,
17    ConvBnReLU3d,
18    ConvReLU1d,
19    ConvReLU2d,
20    ConvReLU3d,
21    freeze_bn_stats,
22    update_bn_stats,
23)
24
25
26__all__ = [
27    # Modules
28    "ConvBn1d",
29    "ConvBnReLU1d",
30    "ConvReLU1d",
31    "ConvBn2d",
32    "ConvBnReLU2d",
33    "ConvReLU2d",
34    "ConvBn3d",
35    "ConvBnReLU3d",
36    "ConvReLU3d",
37    # Utilities
38    "freeze_bn_stats",
39    "update_bn_stats",
40]
41