|
|
@ -168,9 +168,10 @@ void Pool2dOpMaker::Make() {
|
|
|
|
"be ignored."); // TODO(Chengduo): Add checker.
|
|
|
|
"be ignored."); // TODO(Chengduo): Add checker.
|
|
|
|
// (Currently,
|
|
|
|
// (Currently,
|
|
|
|
// TypedAttrChecker don't support vector type.)
|
|
|
|
// TypedAttrChecker don't support vector type.)
|
|
|
|
AddAttr<bool>("global_pooling",
|
|
|
|
AddAttr<bool>(
|
|
|
|
|
|
|
|
"global_pooling",
|
|
|
|
"(bool, default false) Whether to use the global pooling. "
|
|
|
|
"(bool, default false) Whether to use the global pooling. "
|
|
|
|
"If global_pooling = true, ksize and paddings will be ignored.")
|
|
|
|
"If global_pooling = true, kernel size and paddings will be ignored.")
|
|
|
|
.SetDefault(false);
|
|
|
|
.SetDefault(false);
|
|
|
|
AddAttr<std::vector<int>>("strides",
|
|
|
|
AddAttr<std::vector<int>>("strides",
|
|
|
|
"(vector<int>, default {1, 1}), strides(height, "
|
|
|
|
"(vector<int>, default {1, 1}), strides(height, "
|
|
|
@ -182,7 +183,7 @@ void Pool2dOpMaker::Make() {
|
|
|
|
"paddings",
|
|
|
|
"paddings",
|
|
|
|
"(vector<int>, default {0,0}), paddings(height, width) of pooling "
|
|
|
|
"(vector<int>, default {0,0}), paddings(height, width) of pooling "
|
|
|
|
"operator."
|
|
|
|
"operator."
|
|
|
|
"If global_pooling = true, paddings and ksize will be ignored.")
|
|
|
|
"If global_pooling = true, paddings and kernel size will be ignored.")
|
|
|
|
.SetDefault({0, 0});
|
|
|
|
.SetDefault({0, 0});
|
|
|
|
AddAttr<bool>(
|
|
|
|
AddAttr<bool>(
|
|
|
|
"exclusive",
|
|
|
|
"exclusive",
|
|
|
@ -333,7 +334,7 @@ void Pool3dOpMaker::Make() {
|
|
|
|
AddAttr<bool>(
|
|
|
|
AddAttr<bool>(
|
|
|
|
"global_pooling",
|
|
|
|
"global_pooling",
|
|
|
|
"(bool, default false) Whether to use the global pooling. "
|
|
|
|
"(bool, default false) Whether to use the global pooling. "
|
|
|
|
"If global_pooling = true, ksize and paddings will be ignored.")
|
|
|
|
"If global_pooling = true, kernel size and paddings will be ignored.")
|
|
|
|
.SetDefault(false);
|
|
|
|
.SetDefault(false);
|
|
|
|
AddAttr<std::vector<int>>(
|
|
|
|
AddAttr<std::vector<int>>(
|
|
|
|
"strides",
|
|
|
|
"strides",
|
|
|
|