sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add tflite model

This commit is contained in:
jianghui58 2020-08-29 14:09:27 +08:00
parent 81004f5e90
commit fd865743d0
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc
View File

@ -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

1
mindspore/lite/test/models_tflite.cfg
View File

@ -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