!5486 Add Tflite Models

Merge pull request !5486 from jianghui58/master
5 years ago · c9de948543
parent 219b71423b fd865743d0
commit c9de948543
2 changed files with 2 additions and 0 deletions
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc
@ -148,6 +148,7 @@ kernel::LiteKernel *CpuCastFp32KernelCreator(const std::vector<lite::tensor::Ten

 REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_Cast, CpuCastFp32KernelCreator)
 REG_KERNEL(kCPU, kNumberTypeUInt8, PrimitiveType_Cast, CpuCastFp32KernelCreator)
+REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_Cast, CpuCastFp32KernelCreator)
 #ifndef ENABLE_ARM64
 REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Cast, CpuCastFp32KernelCreator)
 #endif
--- a/mindspore/lite/test/models_tflite.cfg
+++ b/mindspore/lite/test/models_tflite.cfg
@ -76,6 +76,7 @@ siteAI_wireless_depress_w.tflite
 siteAI_wireless_restore_w.tflite
 magenta_arbitrary-image-stylization-v1-256_fp16_prediction_1.tflite
 ml_object_detect.tflite
+ml_object_detect_1.tflite
 hiai_cpu_face_emotion.tflite
 hiai_cpu_face_gazing.tflite
 hiai_cpu_face_headpose.tflite