File tree Expand file tree Collapse file tree 3 files changed +12
-6
lines changed
Expand file tree Collapse file tree 3 files changed +12
-6
lines changed Original file line number Diff line number Diff line change @@ -238,12 +238,14 @@ class SliceOpMaker : public framework::OpProtoAndCheckerMaker {
238238 .SetDefault ({});
239239 AddAttr<bool >(" use_mkldnn" ,
240240 " (bool, default false) Only used in mkldnn kernel" )
241- .SetDefault (false );
241+ .SetDefault (false )
242+ .AsExtra ();
242243 AddAttr<std::string>(
243244 " mkldnn_data_type" ,
244245 " (string, default \" float32\" ). Data type of mkldnn kernel" )
245246 .SetDefault (" float32" )
246- .InEnum ({" float32" , " bfloat16" });
247+ .InEnum ({" float32" , " bfloat16" })
248+ .AsExtra ();
247249 AddComment (R"DOC(
248250Slice Operator.
249251
Original file line number Diff line number Diff line change @@ -162,12 +162,14 @@ class SqueezeOpMaker : public framework::OpProtoAndCheckerMaker {
162162 .SetDefault ({});
163163 AddAttr<bool >(" use_mkldnn" ,
164164 " (bool, default false) Only used in mkldnn kernel" )
165- .SetDefault (false );
165+ .SetDefault (false )
166+ .AsExtra ();
166167 AddAttr<std::string>(
167168 " mkldnn_data_type" ,
168169 " (string, default \" float32\" ). Data type of mkldnn kernel" )
169170 .SetDefault (" float32" )
170- .InEnum ({" float32" , " bfloat16" });
171+ .InEnum ({" float32" , " bfloat16" })
172+ .AsExtra ();
171173 AddComment (R"DOC(
172174 Squeeze Operator.
173175
Original file line number Diff line number Diff line change @@ -133,12 +133,14 @@ class TransposeOpMaker : public framework::OpProtoAndCheckerMaker {
133133 " use_quantizer" ,
134134 " (bool, default false) "
135135 " This parameter is no longer used. Use 'mkldnn_data_type' instead." )
136- .SetDefault (false );
136+ .SetDefault (false )
137+ .AsExtra ();
137138 AddAttr<std::string>(
138139 " mkldnn_data_type" ,
139140 " (string, default \" float32\" ). Data type of mkldnn kernel" )
140141 .SetDefault (" float32" )
141- .InEnum ({" float32" , " int8" , " bfloat16" });
142+ .InEnum ({" float32" , " int8" , " bfloat16" })
143+ .AsExtra ();
142144 /* int8 parameters */
143145 AddComment (R"DOC(
144146Transpose Operator.
You can’t perform that action at this time.
0 commit comments