You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
563 B
34 lines
563 B
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
|
|
|
|
|
|
__all__ = [
|
|
"BNReLU2d",
|
|
"BNReLU3d",
|
|
"ConvBn1d",
|
|
"ConvBn2d",
|
|
"ConvBn3d",
|
|
"ConvBnReLU1d",
|
|
"ConvBnReLU2d",
|
|
"ConvBnReLU3d",
|
|
"ConvReLU1d",
|
|
"ConvReLU2d",
|
|
"ConvReLU3d",
|
|
"LinearBn1d",
|
|
"LinearReLU",
|
|
]
|