定义
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};
};
父主题: LinearOperation