提交 9d05738d 编写于 作者: M Megvii Engine Team

Merge pull request #225 from chenjiahui0131:pixel_shuffle_module

GitOrigin-RevId: d94533655a7a765e0dbdc3b26752ec0548e5acda
...@@ -33,6 +33,7 @@ from .lrn import LocalResponseNorm ...@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册