提交 50c21c84 编写于 作者: N NazgulLee 提交者: Yanzhan Yang

fix bug: exclusive param may not exist (#1725)

上级 7a395f69
......@@ -909,7 +909,12 @@ class PoolParam : public OpParam {
paddings_ = GetAttr<vector<int>>("paddings", attrs);
ceil_mode_ = GetAttr<bool>("ceil_mode", attrs);
global_pooling_ = GetAttr<bool>("global_pooling", attrs);
if (HasAttr("exclusive", attrs)) {
exclusive_ = GetAttr<bool>("exclusive", attrs);
} else {
exclusive_ = true;
}
}
const GType *Input() const { return input_; }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册