Skip to content

Commit e44b38e

Browse files
ZihengJiangtqchen
authored andcommitted
[TOPI] Update conv schedule on rasp (#497)
1 parent 81f9d5b commit e44b38e

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

topi/python/topi/nn/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
from .batch_norm import *
66
from .conv2d import *
7-
from .depthwise_convolution import *
7+
from .depthwise_conv2d import *
88
from .elemwise import *
99
from .dilate import *
1010
from .flatten import *

topi/python/topi/rasp/conv2d.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@
3232
SpatialPack(1, 4, 8, 4, 8, False),
3333
SpatialPack(2, 2, 8, 1, 4, False),
3434
SpatialPack(2, 2, 8, 1, 8, False),
35-
SpatialPack(1, 1, 16, 1, 4, False),
36-
SpatialPack(1, 1, 4, 1, 4, True),
35+
Im2ColPack(7, 4, 1, 16, False),
36+
Im2ColPack(7, 4, 1, 4, True),
3737
]
3838

3939
def _schedule_conv2d(wkl):

0 commit comments

Comments
 (0)