!4589 Add REG_KERNEL UInt8

Merge pull request !4589 from jianghui/master

!4589 Add REG_KERNEL UInt8
Merge pull request !4589 from jianghui/master
07b9f0a7 · mindspore-ci-bot · Gitee · a812bf2e · 23abd6a4 · 07b9f0a7
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc +1 -0

未找到文件。
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc
@@ -146,6 +146,7 @@ kernel::LiteKernel *CpuCastFp32KernelCreator(const std::vector<lite::tensor::Ten
 }

 REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_Cast, CpuCastFp32KernelCreator)
+REG_KERNEL(kCPU, kNumberTypeUInt8, PrimitiveType_Cast, CpuCastFp32KernelCreator)
 #ifndef ENABLE_ARM64
 REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Cast, CpuCastFp32KernelCreator)
 #endif