add REG_KERNEL UInt8

23abd6a4 · jianghui58 · f41ca6b5 · 23abd6a4
隐藏空白更改
内联并排

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