未验证 提交 0a100120 编写于 作者: L liu zhengxi 提交者: GitHub

Fix the conv compute shape (#2534)

上级 0349bfd0
...@@ -65,7 +65,7 @@ class Conv2dCompute : public KernelLite<TARGET(kX86), PRECISION(kFloat)> { ...@@ -65,7 +65,7 @@ class Conv2dCompute : public KernelLite<TARGET(kX86), PRECISION(kFloat)> {
col_shape_vec[j + 1 + data_dim] = output_shape_vec[j + 2]; col_shape_vec[j + 1 + data_dim] = output_shape_vec[j + 2];
} }
lite::DDim col_shape(col_shape_vec); 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( bool is_expand = IsExpand(
filter_shape_vec, param.strides, *param.paddings, *param.dilations); filter_shape_vec, param.strides, *param.paddings, *param.dilations);
lite::Tensor col; lite::Tensor col;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册