Browse Source

Merge pull request #225 from chenjiahui0131:pixel_shuffle_module

GitOrigin-RevId: d94533655a
release-1.7
Megvii Engine Team 3 years ago
parent
commit
9d05738d54
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/python/megengine/module/__init__.py

+ 1
- 0
imperative/python/megengine/module/__init__.py View File

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


Loading…
Cancel
Save