Commit e44b38e3 by ziheng Committed by Tianqi Chen

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

parent 81f9d5b3
...@@ -4,7 +4,7 @@ from __future__ import absolute_import as _abs ...@@ -4,7 +4,7 @@ from __future__ import absolute_import as _abs
from .batch_norm import * from .batch_norm import *
from .conv2d import * from .conv2d import *
from .depthwise_convolution import * from .depthwise_conv2d import *
from .elemwise import * from .elemwise import *
from .dilate import * from .dilate import *
from .flatten import * from .flatten import *
......
...@@ -32,8 +32,8 @@ _SCHEDULES = [ ...@@ -32,8 +32,8 @@ _SCHEDULES = [
SpatialPack(1, 4, 8, 4, 8, False), SpatialPack(1, 4, 8, 4, 8, False),
SpatialPack(2, 2, 8, 1, 4, False), SpatialPack(2, 2, 8, 1, 4, False),
SpatialPack(2, 2, 8, 1, 8, False), SpatialPack(2, 2, 8, 1, 8, False),
SpatialPack(1, 1, 16, 1, 4, False), Im2ColPack(7, 4, 1, 16, False),
SpatialPack(1, 1, 4, 1, 4, True), Im2ColPack(7, 4, 1, 4, True),
] ]
def _schedule_conv2d(wkl): def _schedule_conv2d(wkl):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment