昇腾社区首页
中文
注册

定义

struct LinearParam {
    enum MatmulType : uint8_t {
        MATMUL_UNDEFINED = 0,
        MATMUL_EIN_SUM,
    };
    enum QuantMode : uint8_t {      
       QUANT_UNDEFINED = 0,
       PER_CHANNEL,      
       PER_TOKEN   
    };
    bool transposeA = false;
    bool transposeB = true;
    bool hasBias = true;
    aclDataType outDataType = ACL_DT_UNDEFINED;
    bool enAccum = false;
    MatmulType matmulType = MATMUL_UNDEFINED;
    QuantMode quantMode = QUANT_UNDEFINED ;
    uint8_t rsv[21] = {0};
};