File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff 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(
194197Assume feature vectors exist on dimensions
You can’t perform that action at this time.
0 commit comments