提交 dd273f02 编写于 作者: H HappyAngel 提交者: yiicy

[lite] fix compute shape error in operator/conv_op.cc (#2490)

 fix ccompute shape error in ooperators/conv_op.h, test=develop
上级 9df7010b
......@@ -62,9 +62,9 @@ inline void UpdatePaddingAndDilation(std::vector<int>* paddings,
if (padding_algorithm == "SAME") {
for (size_t i = 0; i < strides.size(); ++i) {
int out_size = (data_dims[i + 2] + strides[i] - 1) / strides[i];
int pad_sum =
std::max((out_size - 1) * strides[i] + ksize[i] - data_dims[i + 2],
(int64_t)0);
int pad_sum = std::max(
(out_size - 1) * strides[i] + ksize[i + 2] - data_dims[i + 2],
(int64_t)0);
int pad_0 = pad_sum / 2;
int pad_1 = pad_sum - pad_0;
// pad
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册