fix smart reply

4 years ago · c95e5ec6c4
parent bbfef6233a
commit c95e5ec6c4
2 changed files with 2 additions and 2 deletions
--- a/mindspore/lite/src/runtime/kernel/arm/string/hashtable_lookup.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/string/hashtable_lookup.cc
@ -90,5 +90,5 @@ kernel::LiteKernel *CpuHashtableLookupKernelCreator(const std::vector<lite::Tens
  return kernel;
 }

-REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_HashtableLookup, CpuHashtableLookupKernelCreator)
+REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_HashtableLookup, CpuHashtableLookupKernelCreator)
 }  // namespace mindspore::kernel
--- a/mindspore/lite/src/runtime/kernel/arm/string/predict.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/string/predict.cc
@ -113,5 +113,5 @@ kernel::LiteKernel *CpuPredictKernelCreator(const std::vector<lite::Tensor *> &i
  return kernel;
 }

-REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_CustomPredict, CpuPredictKernelCreator)
+REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_CustomPredict, CpuPredictKernelCreator)
 }  // namespace mindspore::kernel