diff --git a/lite/kernels/arm/lookup_table_compute.cc b/lite/kernels/arm/lookup_table_compute.cc index af9426f3f4a7d9dd0d1260143b7b3e8aea15a034..f024d8c5ea5eb41ef45c5ce0416d967ba7a3651d 100644 --- a/lite/kernels/arm/lookup_table_compute.cc +++ b/lite/kernels/arm/lookup_table_compute.cc @@ -67,22 +67,22 @@ void LookupTableCompute::Run() { REGISTER_LITE_KERNEL(lookup_table, kARM, - kFloat, + kAny, kNCHW, paddle::lite::kernels::arm::LookupTableCompute, def) .BindInput("W", {LiteType::GetTensorTy(TARGET(kARM))}) - .BindInput("Ids", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt64))}) + .BindInput("Ids", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kAny))}) .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))}) .Finalize(); REGISTER_LITE_KERNEL(lookup_table_v2, kARM, - kFloat, + kAny, kNCHW, paddle::lite::kernels::arm::LookupTableCompute, def) .BindInput("W", {LiteType::GetTensorTy(TARGET(kARM))}) - .BindInput("Ids", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt64))}) + .BindInput("Ids", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kAny))}) .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))}) .Finalize(); diff --git a/lite/kernels/arm/lookup_table_compute.h b/lite/kernels/arm/lookup_table_compute.h index 2b66835e71618e299a2f34513dfcb32988848d74..f0c50f55b6d358d8855c78c27f62d53904c3c29d 100644 --- a/lite/kernels/arm/lookup_table_compute.h +++ b/lite/kernels/arm/lookup_table_compute.h @@ -21,7 +21,7 @@ namespace lite { namespace kernels { namespace arm { -class LookupTableCompute : public KernelLite { +class LookupTableCompute : public KernelLite { public: using param_t = operators::LookupTableParam; diff --git a/lite/kernels/arm/lookup_table_compute_test.cc b/lite/kernels/arm/lookup_table_compute_test.cc index 78748edf39c43c5451f8fa3c4d63bde7405c7078..ef6b04862df666cb5c8f3695ca1823cf90c1b313 100644 --- a/lite/kernels/arm/lookup_table_compute_test.cc +++ b/lite/kernels/arm/lookup_table_compute_test.cc @@ -53,7 +53,7 @@ void lookup_table_compute_ref(const operators::LookupTableParam ¶m) { TEST(lookup_table_arm, retrieve_op) { auto lookup_table = - KernelRegistry::Global().Create( + KernelRegistry::Global().Create( "lookup_table"); ASSERT_FALSE(lookup_table.empty()); ASSERT_TRUE(lookup_table.front()); @@ -61,7 +61,7 @@ TEST(lookup_table_arm, retrieve_op) { TEST(lookup_table_arm, init) { LookupTableCompute lookup_table; - ASSERT_EQ(lookup_table.precision(), PRECISION(kFloat)); + ASSERT_EQ(lookup_table.precision(), PRECISION(kAny)); ASSERT_EQ(lookup_table.target(), TARGET(kARM)); } @@ -112,4 +112,4 @@ TEST(lookup_table_arm, compute) { } // namespace lite } // namespace paddle -USE_LITE_KERNEL(lookup_table, kARM, kFloat, kNCHW, def); +USE_LITE_KERNEL(lookup_table, kARM, kAny, kNCHW, def);