From 65d1e46e9f7128c34aff67a0e77a098c35fc4ccc Mon Sep 17 00:00:00 2001 From: ZihengJiang Date: Wed, 27 Sep 2017 20:27:19 -0700 Subject: [PATCH] [TOPI] Update conv schedule on rasp --- topi/python/topi/nn/__init__.py | 2 +- .../topi/nn/{depthwise_convolution.py => depthwise_conv2d.py} | 0 topi/python/topi/rasp/conv2d.py | 4 ++-- 3 files changed, 3 insertions(+), 3 deletions(-) rename topi/python/topi/nn/{depthwise_convolution.py => depthwise_conv2d.py} (100%) diff --git a/topi/python/topi/nn/__init__.py b/topi/python/topi/nn/__init__.py index 514149b00aa0..9e4a5bfbc4a3 100644 --- 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 b/topi/python/topi/nn/depthwise_conv2d.py similarity index 100% rename from topi/python/topi/nn/depthwise_convolution.py rename to topi/python/topi/nn/depthwise_conv2d.py diff --git a/topi/python/topi/rasp/conv2d.py b/topi/python/topi/rasp/conv2d.py index e208a67d8a20..7d07030ad887 100644 --- 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):