提交 dde43023 编写于 作者: W weihaoji

[XPU] bugfix on pooling

test=develop test=xpu
上级 94b0ebd7
......@@ -38,8 +38,7 @@ void Pool2DCompute::Run() {
auto& paddings = *param.paddings;
auto type = xdnn::MAX_WITHOUT_INDEX;
if (param.pooling_type == "avg") {
if (paddings[0] == 0 && paddings[1] == 0 && paddings[2] == 0 &&
paddings[3] == 0) {
if (param.exclusive) {
type = xdnn::AVG_WITHOUT_PAD;
} else {
type = xdnn::AVG_WITH_PAD;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册