!8149 [MSLITE][Develop] fix smart reply

Merge pull request !8149 from sunsuodong/fix_smart_reply
5 years ago · bf640d229a
parent bbfef6233a c95e5ec6c4
commit bf640d229a
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