|
|
|
@@ -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 |