From c95e5ec6c4fa47773632222513507770ac384d4d Mon Sep 17 00:00:00 2001 From: sunsuodong Date: Tue, 3 Nov 2020 10:10:33 +0800 Subject: [PATCH] fix smart reply --- .../lite/src/runtime/kernel/arm/string/hashtable_lookup.cc | 2 +- mindspore/lite/src/runtime/kernel/arm/string/predict.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/string/hashtable_lookup.cc b/mindspore/lite/src/runtime/kernel/arm/string/hashtable_lookup.cc index f3b8e160f7..0f8d4829dd 100644 --- 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 &i return kernel; } -REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_CustomPredict, CpuPredictKernelCreator) +REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_CustomPredict, CpuPredictKernelCreator) } // namespace mindspore::kernel