[TOPI] Update conv schedule on rasp (#497)

apache · Sep 28, 2017 · e44b38e · e44b38e
1 parent 81f9d5b
commit e44b38e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/topi/python/topi/nn/__init__.py b/topi/python/topi/nn/__init__.py
@@ -4,7 +4,7 @@
 
 from .batch_norm import *
 from .conv2d import *
-from .depthwise_convolution import *
+from .depthwise_conv2d import *
 from .elemwise import *
 from .dilate import *
 from .flatten import *

diff --git a/topi/python/topi/nn/depthwise_convolution.py → topi/python/topi/nn/depthwise_conv2d.py b/topi/python/topi/nn/depthwise_convolution.py → topi/python/topi/nn/depthwise_conv2d.py
diff --git a/topi/python/topi/rasp/conv2d.py b/topi/python/topi/rasp/conv2d.py
@@ -32,8 +32,8 @@
     SpatialPack(1, 4, 8, 4, 8, False),
     SpatialPack(2, 2, 8, 1, 4, False),
     SpatialPack(2, 2, 8, 1, 8, False),
-    SpatialPack(1, 1, 16, 1, 4, False),
-    SpatialPack(1, 1, 4, 1, 4, True),
+    Im2ColPack(7, 4, 1, 16, False),
+    Im2ColPack(7, 4, 1, 4, True),
 ]
 
 def _schedule_conv2d(wkl):