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
from .batch_norm import *
from .conv2d import *
from .depthwise_convolution import *
from .depthwise_conv2d import *
from .elemwise import *
from .dilate import *
from .flatten import *
......
......@@ -32,8 +32,8 @@ _SCHEDULES = [
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):
......
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