From 9a1e0fa64b4aa2b6592439892c4478977036ad06 Mon Sep 17 00:00:00 2001 From: sunsuodong Date: Mon, 26 Oct 2020 20:09:33 +0800 Subject: [PATCH] add SquaredDifference fp16 regist --- mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc b/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc index 0ff67245dd..5a4763fecc 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc @@ -266,4 +266,5 @@ REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_LessEqual, CpuArithmeticFp16K REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Greater, CpuArithmeticFp16KernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_GreaterEqual, CpuArithmeticFp16KernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Eltwise, CpuArithmeticFp16KernelCreator) +REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_SquaredDifference, CpuArithmeticFp16KernelCreator) } // namespace mindspore::kernel