|
|
|
@@ -42,7 +42,8 @@ ArithmeticSelfFunc ArithmeticSelfCPUKernel::GetArithmeticSelfFun(int primitive_t |
|
|
|
{mindspore::schema::PrimitiveType_Ceil, ElementCeil}, |
|
|
|
{mindspore::schema::PrimitiveType_Round, ElementRound}, |
|
|
|
{mindspore::schema::PrimitiveType_Neg, ElementNegative}, |
|
|
|
{mindspore::schema::PrimitiveType_Reciprocal, ElementReciprocal}}; |
|
|
|
{mindspore::schema::PrimitiveType_Reciprocal, ElementReciprocal}, |
|
|
|
{mindspore::schema::PrimitiveType_Erf, ElementErf}}; |
|
|
|
for (size_t i = 0; i < sizeof(type_func_table) / sizeof(TYPE_FUNC_INFO); i++) { |
|
|
|
if (type_func_table[i].primitive_type_ == primitive_type) { |
|
|
|
return type_func_table[i].func_; |
|
|
|
|