|
|
|
@@ -40,6 +40,8 @@ const std::map<std::string, OperatorType> DictOpType{ |
|
|
|
{CONV2D, OperatorType::kRecConvolution}, |
|
|
|
{MAXPOOL, OperatorType::kRecPooling}, |
|
|
|
{MAXPOOLV2, OperatorType::kRecPooling}, |
|
|
|
{POOLING, OperatorType::kRecPooling}, |
|
|
|
{MAX_POOL_WITH_ARGMAX, OperatorType::kRecPooling}, |
|
|
|
{SIMPLE_MEAN, OperatorType::kRecPooling}, |
|
|
|
{RESHAPE, OperatorType::kRecReshape}, |
|
|
|
{BIAS_ADD, OperatorType::kRecBiasAdd}, |
|
|
|
@@ -56,7 +58,8 @@ const std::map<std::string, OperatorType> DictOpType{ |
|
|
|
{GATHERV2, OperatorType::kRecGatherV2}, |
|
|
|
{ARGMAXWITHVALUE, OperatorType::kRecArgWithValue}, |
|
|
|
{ARGMINWITHVALUE, OperatorType::kRecArgWithValue}, |
|
|
|
|
|
|
|
// Activation OP |
|
|
|
{ACTIVATION, OperatorType::kRecReLU}, |
|
|
|
{RELU, OperatorType::kRecReLU}, |
|
|
|
{"ReLU6", OperatorType::kRecReLU}, |
|
|
|
{"ReLUV2", OperatorType::kRecReLU}, |
|
|
|
@@ -65,9 +68,11 @@ const std::map<std::string, OperatorType> DictOpType{ |
|
|
|
{"HSigmoid", OperatorType::kRecReLU}, |
|
|
|
{GELU, OperatorType::kRecReLU}, |
|
|
|
{TANH, OperatorType::kRecReLU}, |
|
|
|
|
|
|
|
{SOFTPLUS, OperatorType::kRecReLU}, |
|
|
|
{SOFTSIGN, OperatorType::kRecReLU}, |
|
|
|
{PRELU, OperatorType::kRecPReLU}, |
|
|
|
|
|
|
|
// Elm-wise OP |
|
|
|
{TRANSPOSE, OperatorType::kRecElmWiseOp}, |
|
|
|
{TRANSPOSE, OperatorType::kRecElmWiseOp}, |
|
|
|
{L2_NORMALIZE, OperatorType::kRecElmWiseOp}, |
|
|
|
{TENSOR_ADD, OperatorType::kRecElmWiseOp}, |
|
|
|
@@ -85,6 +90,12 @@ const std::map<std::string, OperatorType> DictOpType{ |
|
|
|
{LOG, OperatorType::kRecElmWiseOp}, |
|
|
|
{COS, OperatorType::kRecElmWiseOp}, |
|
|
|
{ACOS, OperatorType::kRecElmWiseOp}, |
|
|
|
{ASIN, OperatorType::kRecElmWiseOp}, |
|
|
|
{ASINH, OperatorType::kRecElmWiseOp}, |
|
|
|
{ATAN, OperatorType::kRecElmWiseOp}, |
|
|
|
{ATANH, OperatorType::kRecElmWiseOp}, |
|
|
|
{EXPM1, OperatorType::kRecElmWiseOp}, |
|
|
|
{LOG1P, OperatorType::kRecElmWiseOp}, |
|
|
|
{LOGICALNOT, OperatorType::kRecElmWiseOp}, |
|
|
|
{"LogicalAnd", OperatorType::kRecElmWiseOp}, |
|
|
|
{"LogicalOr", OperatorType::kRecElmWiseOp}, |
|
|
|
@@ -94,8 +105,12 @@ const std::map<std::string, OperatorType> DictOpType{ |
|
|
|
{"AddN", OperatorType::kRecElmWiseOp}, |
|
|
|
{"AccumulateNV2", OperatorType::kRecElmWiseOp}, |
|
|
|
{"Atan2", OperatorType::kRecElmWiseOp}, |
|
|
|
{"Erf", OperatorType::kRecElmWiseOp}, |
|
|
|
{"Floor", OperatorType::kRecElmWiseOp}, |
|
|
|
{ELU, OperatorType::kRecElmWiseOp}, |
|
|
|
{ERF, OperatorType::kRecElmWiseOp}, |
|
|
|
{ERFC, OperatorType::kRecElmWiseOp}, |
|
|
|
{MOD, OperatorType::kRecElmWiseOp}, |
|
|
|
{FLOOR, OperatorType::kRecElmWiseOp}, |
|
|
|
{CEIL, OperatorType::kRecElmWiseOp}, |
|
|
|
{FLOORDIV, OperatorType::kRecElmWiseOp}, |
|
|
|
{"FloorMod", OperatorType::kRecElmWiseOp}, |
|
|
|
{GREATER, OperatorType::kRecElmWiseOp}, |
|
|
|
@@ -107,12 +122,22 @@ const std::map<std::string, OperatorType> DictOpType{ |
|
|
|
{MINIMUM, OperatorType::kRecElmWiseOp}, |
|
|
|
{EQUAL, OperatorType::kRecElmWiseOp}, |
|
|
|
{NOT_EQUAL, OperatorType::kRecElmWiseOp}, |
|
|
|
{APPROXIMATEEQUAL, OperatorType::kRecElmWiseOp}, |
|
|
|
{INV, OperatorType::kRecElmWiseOp}, |
|
|
|
{BESSELI0E, OperatorType::kRecElmWiseOp}, |
|
|
|
{BESSELI1E, OperatorType::kRecElmWiseOp}, |
|
|
|
{ZEROSLIKE, OperatorType::kRecElmWiseOp}, |
|
|
|
{ONESLIKE, OperatorType::kRecElmWiseOp}, |
|
|
|
{DIVNONAN, OperatorType::kRecElmWiseOp}, |
|
|
|
{"Reciprocal", OperatorType::kRecElmWiseOp}, |
|
|
|
{"Round", OperatorType::kRecElmWiseOp}, |
|
|
|
{"Rsqrt", OperatorType::kRecElmWiseOp}, |
|
|
|
{"Sign", OperatorType::kRecElmWiseOp}, |
|
|
|
{"Sin", OperatorType::kRecElmWiseOp}, |
|
|
|
{SIN, OperatorType::kRecElmWiseOp}, |
|
|
|
{SINH, OperatorType::kRecElmWiseOp}, |
|
|
|
{TAN, OperatorType::kRecElmWiseOp}, |
|
|
|
{ASSIGN, OperatorType::kRecElmWiseOp}, |
|
|
|
{ASSIGN_ADD, OperatorType::kRecElmWiseOp}, |
|
|
|
{ASSIGN_SUB, OperatorType::kRecElmWiseOp}, |
|
|
|
{"AssignAdd", OperatorType::kRecElmWiseOp}}; |
|
|
|
|
|
|
|
|