xref: /aosp_15_r20/external/pytorch/torch/nn/intrinsic/modules/fused.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1from torch.ao.nn.intrinsic import (
2    BNReLU2d,
3    BNReLU3d,
4    ConvBn1d,
5    ConvBn2d,
6    ConvBn3d,
7    ConvBnReLU1d,
8    ConvBnReLU2d,
9    ConvBnReLU3d,
10    ConvReLU1d,
11    ConvReLU2d,
12    ConvReLU3d,
13    LinearBn1d,
14    LinearReLU,
15)
16from torch.ao.nn.intrinsic.modules.fused import _FusedModule  # noqa: F401
17
18
19__all__ = [
20    "BNReLU2d",
21    "BNReLU3d",
22    "ConvBn1d",
23    "ConvBn2d",
24    "ConvBn3d",
25    "ConvBnReLU1d",
26    "ConvBnReLU2d",
27    "ConvBnReLU3d",
28    "ConvReLU1d",
29    "ConvReLU2d",
30    "ConvReLU3d",
31    "LinearBn1d",
32    "LinearReLU",
33]
34