提交 6b393f96 编写于 作者: L liu zhengxi 提交者: GitHub

Fix the conv compute shape (#2534)

上级 e7ed7286
......@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册