Skip to content

P
Paddle-Lite

PaddlePaddle / Paddle-Lite

通知 331
Star 4
Fork 1
P
Paddle-Lite
未验证 提交 68d75d33 编写于 作者: H huzhiqiang 提交者： GitHub
浏览文件
操作

【BUG FIX】Fix lookup_table OP(#2774)

上级 cb0ce132
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
lite/kernels/arm/lookup_table_compute.cc
浏览文件 @ 68d75d33
...@@ -72,7 +72,7 @@ REGISTER_LITE_KERNEL(lookup_table, ...@@ -72,7 +72,7 @@ REGISTER_LITE_KERNEL(lookup_table,
paddle::lite::kernels::arm::LookupTableCompute, paddle::lite::kernels::arm::LookupTableCompute,
def) def)
.BindInput("W", {LiteType::GetTensorTy(TARGET(kARM))}) .BindInput("W", {LiteType::GetTensorTy(TARGET(kARM))})
.BindInput("Ids", {LiteType::GetTensorTy(TARGET(kARM))}) .BindInput("Ids", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt64))})
.BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))}) .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))})
.Finalize(); .Finalize();
...@@ -83,6 +83,6 @@ REGISTER_LITE_KERNEL(lookup_table_v2, ...@@ -83,6 +83,6 @@ REGISTER_LITE_KERNEL(lookup_table_v2,
paddle::lite::kernels::arm::LookupTableCompute, paddle::lite::kernels::arm::LookupTableCompute,
def) def)
.BindInput("W", {LiteType::GetTensorTy(TARGET(kARM))}) .BindInput("W", {LiteType::GetTensorTy(TARGET(kARM))})
.BindInput("Ids", {LiteType::GetTensorTy(TARGET(kARM))}) .BindInput("Ids", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt64))})
.BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))}) .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))})
.Finalize(); .Finalize();
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或