Skip to content

Commit e7642ac

Browse files
feng_shuaiAnnaTrainingG
authored andcommitted
transpose/slice/stride_slice/squeeze/unsqueeze op_def_enhance-1.0 (PaddlePaddle#35391)
* transpose/slice/stride_slice/squeeze/unsqueeze op_def_enhance-1.0 * delete infer_flags and decrease-axis * delet infer_flags and decrea_axis
1 parent bf1f5bf commit e7642ac

File tree

3 files changed

+12
-6
lines changed

3 files changed

+12
-6
lines changed

paddle/fluid/operators/slice_op.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff 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(
248250
Slice Operator.
249251

paddle/fluid/operators/squeeze_op.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff 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

paddle/fluid/operators/transpose_op.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff 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(
144146
Transpose Operator.

0 commit comments

Comments
 (0)