From 24597f0a07e4f1424dd9f9f279da8dc154a59af1 Mon Sep 17 00:00:00 2001 From: Cody Yu Date: Fri, 5 Jun 2020 10:27:57 -0700 Subject: [PATCH] Fix reshape usage in ARM Winograd (#5732) --- topi/python/topi/arm_cpu/conv2d_alter_op.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/topi/python/topi/arm_cpu/conv2d_alter_op.py b/topi/python/topi/arm_cpu/conv2d_alter_op.py index 221ccce..3206168 100644 --- a/topi/python/topi/arm_cpu/conv2d_alter_op.py +++ b/topi/python/topi/arm_cpu/conv2d_alter_op.py @@ -117,7 +117,7 @@ def _alter_conv2d_layout(attrs, inputs, tinfos, out_type): weight_expr = relay.reshape(weight_expr, newshape=(KH + tile_size - 1, KW + tile_size - 1, - idxd(CO, VC), VC, CI)) + CO // VC, VC, CI)) weight_expr = relay.transpose(weight_expr, axes=[0, 1, 2, 4, 3]) new_attrs['tile_size'] = tile_size -- 2.7.4