Skip to content

Commit 2ab2ebc

Browse files
committed
add AsExtra() mark for layer_norm
1 parent 67ed7e1 commit 2ab2ebc

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

paddle/fluid/operators/layer_norm_op.cc

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -179,16 +179,19 @@ class LayerNormOpMaker : public framework::OpProtoAndCheckerMaker {
179179
});
180180
AddAttr<bool>("use_mkldnn",
181181
"(bool, default false) Only used in mkldnn kernel")
182-
.SetDefault(false);
182+
.SetDefault(false)
183+
.AsExtra();
183184
AddAttr<std::string>(
184185
"mkldnn_data_type",
185186
"(string, default \"float32\"). Data type of mkldnn kernel")
186187
.SetDefault("float32")
187-
.InEnum({"float32", "bfloat16"});
188+
.InEnum({"float32", "bfloat16"})
189+
.AsExtra();
188190
AddAttr<bool>("is_test",
189191
"(bool, default false) Set to true for inference only, false "
190192
"for training. Some layers may run faster when this is true.")
191-
.SetDefault(false);
193+
.SetDefault(false)
194+
.AsExtra();
192195

193196
AddComment(R"DOC(
194197
Assume feature vectors exist on dimensions

0 commit comments

Comments
 (0)