Browse Source

!7660 [AutoParallel] Support new operators in double recursion

Merge pull request !7660 from Chong/CheckParserInfo
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
77e8ff9b5a
1 changed files with 31 additions and 6 deletions
  1. +31
    -6
      mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/rec_parse_graph.h

+ 31
- 6
mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/rec_parse_graph.h View File

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



Loading…
Cancel
Save