Skip to content

Commit ef7bc36

Browse files
authored
Gather op (#35353)
* add lookup_table_v2_op AsExtra * add grap attr asextra * add gather_op asExtra()
1 parent fc8d46c commit ef7bc36

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddle/fluid/operators/gather_op.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,8 @@ class GatherOpMaker : public framework::OpProtoAndCheckerMaker {
138138
"In backward process, calc the grad when has same index,"
139139
"If true, update the grad using the overwrite mode in same index,"
140140
"If false, using the accumulate mode in same index.")
141-
.SetDefault(true);
141+
.SetDefault(true)
142+
.AsExtra();
142143
AddAttr<int>(
143144
"axis",
144145
"The Tensor which contains the axis that we do gather operation.")

0 commit comments

Comments
 (0)