From 860ddb03ecdc923dae2ecc35375c8b910bfea26e Mon Sep 17 00:00:00 2001 From: sunsuodong Date: Fri, 4 Sep 2020 16:20:04 +0800 Subject: [PATCH] add_eltwise_register --- mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc | 2 ++ 1 file changed, 2 insertions(+) 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 4c0e7e0f1b..34a13724a5 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp16/arithmetic_fp16.cc @@ -45,6 +45,7 @@ using mindspore::schema::PrimitiveType_Mul; using mindspore::schema::PrimitiveType_NotEqual; using mindspore::schema::PrimitiveType_SquaredDifference; using mindspore::schema::PrimitiveType_Sub; +using mindspore::schema::PrimitiveType_Eltwise; namespace mindspore::kernel { ARITHMETIC_FUNC_INFO_FP16 arithmetic_fun_table_fp16[] = { @@ -267,4 +268,5 @@ REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Less, CpuArithmeticFp16Kernel REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_LessEqual, CpuArithmeticFp16KernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Greater, CpuArithmeticFp16KernelCreator) REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_GreaterEqual, CpuArithmeticFp16KernelCreator) +REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Eltwise, CpuArithmeticFp16KernelCreator) } // namespace mindspore::kernel