Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

add tflite model

tags/v1.0.0
jianghui58 5 years ago
parent
81004f5e90
commit
fd865743d0
2 changed files with 2 additions and 0 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc
  2. +1
    -0
      mindspore/lite/test/models_tflite.cfg

+ 1
- 0
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, kNumberTypeFloat32, PrimitiveType_Cast, CpuCastFp32KernelCreator)
REG_KERNEL(kCPU, kNumberTypeUInt8, PrimitiveType_Cast, CpuCastFp32KernelCreator) REG_KERNEL(kCPU, kNumberTypeUInt8, PrimitiveType_Cast, CpuCastFp32KernelCreator)
REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_Cast, CpuCastFp32KernelCreator)
#ifndef ENABLE_ARM64 #ifndef ENABLE_ARM64
REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Cast, CpuCastFp32KernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Cast, CpuCastFp32KernelCreator)
#endif #endif


+ 1
- 0
mindspore/lite/test/models_tflite.cfg View File

@@ -76,6 +76,7 @@ siteAI_wireless_depress_w.tflite
siteAI_wireless_restore_w.tflite siteAI_wireless_restore_w.tflite
magenta_arbitrary-image-stylization-v1-256_fp16_prediction_1.tflite magenta_arbitrary-image-stylization-v1-256_fp16_prediction_1.tflite
ml_object_detect.tflite ml_object_detect.tflite
ml_object_detect_1.tflite
hiai_cpu_face_emotion.tflite hiai_cpu_face_emotion.tflite
hiai_cpu_face_gazing.tflite hiai_cpu_face_gazing.tflite
hiai_cpu_face_headpose.tflite hiai_cpu_face_headpose.tflite


Write Preview
Loading…
Cancel
Save