|
@@ -33,6 +33,7 @@ from .lrn import LocalResponseNorm |
|
|
from .module import Module |
|
|
from .module import Module |
|
|
from .normalization import GroupNorm, InstanceNorm, LayerNorm |
|
|
from .normalization import GroupNorm, InstanceNorm, LayerNorm |
|
|
from .padding import Pad |
|
|
from .padding import Pad |
|
|
|
|
|
from .pixel_shuffle import PixelShuffle |
|
|
from .pooling import AvgPool2d, MaxPool2d |
|
|
from .pooling import AvgPool2d, MaxPool2d |
|
|
from .quant_dequant import DequantStub, QuantStub |
|
|
from .quant_dequant import DequantStub, QuantStub |
|
|
from .sequential import Sequential |
|
|
from .sequential import Sequential |
|
|