mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
b5c006acac
Part of #123062 - #123062 Pull Request resolved: https://github.com/pytorch/pytorch/pull/128865 Approved by: https://github.com/ezyang
37 lines
697 B
Python
37 lines
697 B
Python
from torch.ao.nn.intrinsic import (
|
|
BNReLU2d,
|
|
BNReLU3d,
|
|
ConvBn1d,
|
|
ConvBn2d,
|
|
ConvBn3d,
|
|
ConvBnReLU1d,
|
|
ConvBnReLU2d,
|
|
ConvBnReLU3d,
|
|
ConvReLU1d,
|
|
ConvReLU2d,
|
|
ConvReLU3d,
|
|
LinearBn1d,
|
|
LinearReLU,
|
|
)
|
|
from torch.ao.nn.intrinsic.modules.fused import _FusedModule # noqa: F401
|
|
|
|
# Include the subpackages in case user imports from it directly
|
|
from torch.nn.intrinsic import modules, qat, quantized # noqa: F401
|
|
|
|
|
|
__all__ = [
|
|
"ConvBn1d",
|
|
"ConvBn2d",
|
|
"ConvBn3d",
|
|
"ConvBnReLU1d",
|
|
"ConvBnReLU2d",
|
|
"ConvBnReLU3d",
|
|
"ConvReLU1d",
|
|
"ConvReLU2d",
|
|
"ConvReLU3d",
|
|
"LinearReLU",
|
|
"BNReLU2d",
|
|
"BNReLU3d",
|
|
"LinearBn1d",
|
|
]
|