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

add lookup_table_v2_op AsExtra (#35265)

* add lookup_table_v2_op AsExtra

* add grap attr asextra
上级 4fe1bb45
...@@ -81,10 +81,12 @@ class LookupTableV2OpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -81,10 +81,12 @@ class LookupTableV2OpMaker : public framework::OpProtoAndCheckerMaker {
AddAttr<bool>("is_sparse", AddAttr<bool>("is_sparse",
"(boolean, default false) " "(boolean, default false) "
"Sparse update.") "Sparse update.")
.SetDefault(false); .SetDefault(false)
.AsExtra();
AddAttr<bool>("is_distributed", AddAttr<bool>("is_distributed",
"(boolean, default false) distributed lookup table.") "(boolean, default false) distributed lookup table.")
.SetDefault(false); .SetDefault(false)
.AsExtra();
AddAttr<int64_t>("padding_idx", AddAttr<int64_t>("padding_idx",
"(int64, default -1) " "(int64, default -1) "
"If the value is -1, it makes no effect to lookup. " "If the value is -1, it makes no effect to lookup. "
...@@ -93,22 +95,27 @@ class LookupTableV2OpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -93,22 +95,27 @@ class LookupTableV2OpMaker : public framework::OpProtoAndCheckerMaker {
.SetDefault(kNoPadding); .SetDefault(kNoPadding);
// for parameter prefetch // for parameter prefetch
AddAttr<bool>("remote_prefetch", "").SetDefault(false); AddAttr<bool>("remote_prefetch", "").SetDefault(false).AsExtra();
AddAttr<int>("trainer_id", "trainer id from 0 ~ worker_num.").SetDefault(0); AddAttr<int>("trainer_id", "trainer id from 0 ~ worker_num.")
.SetDefault(0)
.AsExtra();
AddAttr<std::vector<int64_t>>("height_sections", AddAttr<std::vector<int64_t>>("height_sections",
"Height for each output SelectedRows.") "Height for each output SelectedRows.")
.SetDefault(std::vector<int64_t>({})); .SetDefault(std::vector<int64_t>({}))
.AsExtra();
AddAttr<std::vector<std::string>>( AddAttr<std::vector<std::string>>(
"epmap", "epmap",
"(string vector, default 127.0.0.1:6164)" "(string vector, default 127.0.0.1:6164)"
"Server endpoints in the order of input variables for mapping") "Server endpoints in the order of input variables for mapping")
.SetDefault({}); .SetDefault({})
.AsExtra();
AddAttr<std::vector<std::string>>( AddAttr<std::vector<std::string>>(
"table_names", "table_names",
"(string vector, the split table names that will be fetched from " "(string vector, the split table names that will be fetched from "
"parameter server)" "parameter server)"
"in the order of input variables for mapping") "in the order of input variables for mapping")
.SetDefault({}); .SetDefault({})
.AsExtra();
AddComment(R"DOC( AddComment(R"DOC(
Lookup Table V2 Operator. Lookup Table V2 Operator.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册