m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix dataType in C++ comment in embedding op (#20004)

Browse Source
fix-python-transpose
Aurelius84 5 years ago committed by GitHub
parent bcb2903e60
commit 2b5b4b3c5e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

2
paddle/fluid/operators/lookup_table_op.cc
Unescape View File

@ -68,7 +68,7 @@ class LookupTableOpMaker : public framework::OpProtoAndCheckerMaker {
"(Tensor) The input represents embedding tensors, " "(Tensor) The input represents embedding tensors, "
"which is a learnable parameter."); "which is a learnable parameter.");
AddInput("Ids", AddInput("Ids",
"An input with type int32 or int64 " "An input with type int64 "
"contains the ids to be looked up in W. " "contains the ids to be looked up in W. "
"The last dimension size must be 1."); "The last dimension size must be 1.");
AddOutput("Out", "The lookup results, which have the same type as W."); AddOutput("Out", "The lookup results, which have the same type as W.");

2
paddle/fluid/operators/lookup_table_v2_op.cc
Unescape View File

@ -65,7 +65,7 @@ class LookupTableV2OpMaker : public framework::OpProtoAndCheckerMaker {
"(Tensor) The input represents embedding tensors, " "(Tensor) The input represents embedding tensors, "
"which is a learnable parameter."); "which is a learnable parameter.");
AddInput("Ids", AddInput("Ids",
"An input with type int32 or int64 " "An input with type int64 "
"contains the ids to be looked up in W. " "contains the ids to be looked up in W. "
"The last dimension size must be 1."); "The last dimension size must be 1.");
AddOutput("Out", "The lookup results, which have the same type as W."); AddOutput("Out", "The lookup results, which have the same type as W.");

Write Preview
Loading…
Cancel
Save