[ARM] lookup_table_v2 supports int32_t ids and int8_t weights (#9678)

* add_dequantize_log test=develop

* [ARM] lookup_table_v2 supports int32_t ids and int8_t weights

Change-Id: I47bbbc9fc490e86cc0e0b2f8be942d077d9bd789

Co-authored-by: sprouteer <[email protected]>

Author: hong19860320

lite/kernels/arm/lookup_table_compute.cc

@@ -91,6 +91,8 @@ using LookupTableInt8Int64 =
     paddle::lite::kernels::arm::LookupTableCompute<int8_t, int64_t>;
 using LookupTableFloatInt32 =
     paddle::lite::kernels::arm::LookupTableCompute<float, int32_t>;
+using LookupTableInt8Int32 =
+    paddle::lite::kernels::arm::LookupTableCompute<int8_t, int32_t>;
 
 REGISTER_LITE_KERNEL(
     lookup_table, kARM, kAny, kNCHW, LookupTableFloatInt64, def)
@@ -115,6 +117,14 @@ REGISTER_LITE_KERNEL(
     .BindPaddleOpVersion("lookup_table_v2", 1)
     .Finalize();
 
+REGISTER_LITE_KERNEL(
+    lookup_table_v2, kARM, kAny, kNCHW, LookupTableInt8Int32, int8_int32)
+    .BindInput("W", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt8))})
+    .BindInput("Ids", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt32))})
+    .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt8))})
+    .BindPaddleOpVersion("lookup_table_v2", 1)
+    .Finalize();
+
 REGISTER_LITE_KERNEL(
     lookup_table, kARM, kAny, kNCHW, LookupTableFloatInt32, float_int32)
     .BindInput("W", {LiteType::GetTensorTy(TARGET(kARM))})