diff --git a/paddle/fluid/operators/gather_op.cc b/paddle/fluid/operators/gather_op.cc index ea28c204ec9cf9e63f1dace5c4a9188b0f1c1719..d35b066be85e78a58303535f72bc9ede3cd3a5e8 100644 --- a/paddle/fluid/operators/gather_op.cc +++ b/paddle/fluid/operators/gather_op.cc @@ -138,7 +138,8 @@ class GatherOpMaker : public framework::OpProtoAndCheckerMaker { "In backward process, calc the grad when has same index," "If true, update the grad using the overwrite mode in same index," "If false, using the accumulate mode in same index.") - .SetDefault(true); + .SetDefault(true) + .AsExtra(); AddAttr( "axis", "The Tensor which contains the axis that we do gather operation.")