定义
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | struct ElewiseParam { enum ElewiseType: int { ELEWISE_UNDEFINED = 0, ELEWISE_CAST, ELEWISE_MULS, ELEWISE_COS, ELEWISE_SIN, ELEWISE_NEG, ELEWISE_QUANT, ELEWISE_LOGICAL_NOT, ELEWISE_ADD, ELEWISE_MUL, ELEWISE_REALDIV, ELEWISE_LOGICAL_AND, ELEWISE_LOGICAL_OR, ELEWISE_LESS, ELEWISE_GREATER, ELEWISE_SUB, ELEWISE_EQUAL, ELEWISE_QUANT_PER_CHANNEL, ELEWISE_DEQUANT_PER_CHANNEL, ELEWISE_DYNAMIC_QUANT, ELEWISE_TANH, ELEWISE_TYPE_MAX }; struct QuantParam { float inputScale = 1.0f; bool asymmetric = false; int inputOffset = 0; uint8_t rsv[20] = {0}; }; struct MulsParam { float varAttr = 0.0f; uint8_t rsv[12] = {0}; }; ElewiseType elewiseType = ELEWISE_UNDEFINED; QuantParam quantParam; MulsParam mulsParam; aclDataType outTensorType = ACL_DT_UNDEFINED; uint8_t rsv[8] = {0}; }; |