From 0a100120fdb142ada65773eae9cba149e9b079c1 Mon Sep 17 00:00:00 2001
From: liu zhengxi <380185688@qq.com>
Date: Mon, 2 Dec 2019 10:00:52 +0800
Subject: [PATCH] Fix the conv compute shape (#2534)

---
 lite/kernels/x86/conv_compute.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lite/kernels/x86/conv_compute.h b/lite/kernels/x86/conv_compute.h
index 063e66a1c1..e9f403059f 100644
--- a/lite/kernels/x86/conv_compute.h
+++ b/lite/kernels/x86/conv_compute.h
@@ -65,7 +65,7 @@ class Conv2dCompute : public KernelLite<TARGET(kX86), PRECISION(kFloat)> {
       col_shape_vec[j + 1 + data_dim] = output_shape_vec[j + 2];
     }
     lite::DDim col_shape(col_shape_vec);
-    lite::DDim col_matrix_shape = col_shape.Flatten2D(data_dim);
+    lite::DDim col_matrix_shape = col_shape.Flatten2D(data_dim + 1);
     bool is_expand = IsExpand(
         filter_shape_vec, param.strides, *param.paddings, *param.dilations);
     lite::Tensor col;
-- 
GitLab