diff --git a/paddle/fluid/operators/pool_op.cc b/paddle/fluid/operators/pool_op.cc index feb47a73ee4059f21fdb824443e3d48f021db8a4..03766ebb465f0d3abbb8601f830d971ccd2d57d0 100644 --- a/paddle/fluid/operators/pool_op.cc +++ b/paddle/fluid/operators/pool_op.cc @@ -336,7 +336,8 @@ void Pool2dOpMaker::Make() { AddAttr( "use_cudnn", "(bool) Only used in cudnn kernel, need install cudnn. Default False") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddAttr( "ceil_mode", "(bool) Whether to use the ceil function to calculate " @@ -345,17 +346,20 @@ void Pool2dOpMaker::Make() { .SetDefault(false); AddAttr("use_mkldnn", "(bool) Only used in mkldnn kernel. Default False") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddAttr( "use_quantizer", "(bool, default false) " "This parameter is no longer used. Use 'mkldnn_data_type' instead.") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddAttr( "mkldnn_data_type", "(string, default \"float32\"). Data type of mkldnn kernel") .SetDefault("float32") - .InEnum({"float32", "int8", "bfloat16"}); + .InEnum({"float32", "int8", "bfloat16"}) + .AsExtra(); AddAttr( "data_format", "(string, default NCHW) Only used in " @@ -366,7 +370,8 @@ void Pool2dOpMaker::Make() { AddAttr("is_test", "(bool, default false) Set to true for inference only, false " "for training. Some layers may run faster when this is true.") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddAttr( "padding_algorithm", @@ -540,7 +545,8 @@ void Pool3dOpMaker::Make() { AddAttr( "use_cudnn", "(bool) Only used in cudnn kernel, need install cudnn. Default False") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddAttr( "ceil_mode", "(bool) Whether to use the ceil function to calculate " @@ -549,7 +555,8 @@ void Pool3dOpMaker::Make() { .SetDefault(false); AddAttr("use_mkldnn", "(bool) Only used in mkldnn kernel. Default False") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddAttr( "data_format", "(string, default NCDHW) Only used in "