未验证 提交 2b5b4b3c 编写于 作者: A Aurelius84 提交者: GitHub

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

上级 bcb2903e
...@@ -68,7 +68,7 @@ class LookupTableOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -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.");
......
...@@ -65,7 +65,7 @@ class LookupTableV2OpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -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.");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册