未验证 提交 ef7bc367 编写于 作者: 沉潜的鱼儿's avatar 沉潜的鱼儿 提交者: GitHub

Gather op (#35353)

* add lookup_table_v2_op AsExtra

* add grap attr asextra

* add gather_op asExtra()
上级 fc8d46c0
...@@ -138,7 +138,8 @@ class GatherOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -138,7 +138,8 @@ class GatherOpMaker : public framework::OpProtoAndCheckerMaker {
"In backward process, calc the grad when has same index," "In backward process, calc the grad when has same index,"
"If true, update the grad using the overwrite mode in same index," "If true, update the grad using the overwrite mode in same index,"
"If false, using the accumulate mode in same index.") "If false, using the accumulate mode in same index.")
.SetDefault(true); .SetDefault(true)
.AsExtra();
AddAttr<int>( AddAttr<int>(
"axis", "axis",
"The Tensor which contains the axis that we do gather operation.") "The Tensor which contains the axis that we do gather operation.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册