[TOPI] Fix tuple unpacking in conv2d int8 compute (apache#13566)

* Remove mutable defaults in mlp_model * Fix tuple unpack in conv2d nchw int8
mikeseven · Dec 29, 2022 · 9142c7c · 9142c7c
1 parent 95bb32c
commit 9142c7c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/topi/arm_cpu/conv2d_int8.py b/python/tvm/topi/arm_cpu/conv2d_int8.py
@@ -57,7 +57,7 @@ def conv2d_NCHWc_int8(cfg, data, kernel, strides, padding, dilation, layout, out
         n, ic_chunk, ih, iw, ic_bn = get_const_tuple(data.shape)
         in_channel = ic_chunk * ic_bn
 
-        oc_chunk, ic_chunk, kh, kw, ic_bn, oc_bn, _ = get_const_tuple(kernel.shape)
+        oc_chunk, ic_chunk, kh, kw, ic_bn, oc_bn = get_const_tuple(kernel.shape)[:6]
         num_filter = oc_chunk * oc_bn
     else:
         # data is nchw, implicitly treat it as nchw1c