diff --git a/paddle/fluid/operators/lookup_table_v2_op.cc b/paddle/fluid/operators/lookup_table_v2_op.cc index f1bb9a985f4c1da262202a98b15847d85ef8e305..2bda2eb889123bc81fa7f829095cda8764ff6e91 100644 --- a/paddle/fluid/operators/lookup_table_v2_op.cc +++ b/paddle/fluid/operators/lookup_table_v2_op.cc @@ -81,10 +81,12 @@ class LookupTableV2OpMaker : public framework::OpProtoAndCheckerMaker { AddAttr("is_sparse", "(boolean, default false) " "Sparse update.") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddAttr("is_distributed", "(boolean, default false) distributed lookup table.") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddAttr("padding_idx", "(int64, default -1) " "If the value is -1, it makes no effect to lookup. " @@ -93,22 +95,27 @@ class LookupTableV2OpMaker : public framework::OpProtoAndCheckerMaker { .SetDefault(kNoPadding); // for parameter prefetch - AddAttr("remote_prefetch", "").SetDefault(false); - AddAttr("trainer_id", "trainer id from 0 ~ worker_num.").SetDefault(0); + AddAttr("remote_prefetch", "").SetDefault(false).AsExtra(); + AddAttr("trainer_id", "trainer id from 0 ~ worker_num.") + .SetDefault(0) + .AsExtra(); AddAttr>("height_sections", "Height for each output SelectedRows.") - .SetDefault(std::vector({})); + .SetDefault(std::vector({})) + .AsExtra(); AddAttr>( "epmap", "(string vector, default 127.0.0.1:6164)" "Server endpoints in the order of input variables for mapping") - .SetDefault({}); + .SetDefault({}) + .AsExtra(); AddAttr>( "table_names", "(string vector, the split table names that will be fetched from " "parameter server)" "in the order of input variables for mapping") - .SetDefault({}); + .SetDefault({}) + .AsExtra(); AddComment(R"DOC( Lookup Table V2 Operator.