From ef7bc3674b59e3484cbc1a97e13710c7f44744ed Mon Sep 17 00:00:00 2001
From: June Weng <hopewj@163.com>
Date: Fri, 3 Sep 2021 11:35:13 +0800
Subject: [PATCH] Gather op (#35353)

* add lookup_table_v2_op AsExtra

* add grap attr asextra

* add gather_op asExtra()
---
 paddle/fluid/operators/gather_op.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/gather_op.cc b/paddle/fluid/operators/gather_op.cc
index ea28c204ec..d35b066be8 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<int>(
         "axis",
         "The Tensor which contains the axis that we do gather operation.")
-- 
GitLab