SetAttr
函数原型
Operator& SetAttr(const string& name, AttrValue&& attrValue);
Operator& SetAttr(const string& name, const UsrQuantizeFactorParams& attrValue);
功能说明
设置算子属性的属性值。
参数说明
参数名 |
输入/输出 |
类型 |
描述 |
---|---|---|---|
name |
输入 |
const string& |
属性名称。 |
attrValue |
输入 |
AttrValue&& |
需设置的属性值。 |
attrValue |
输入 |
const UsrQuantizeFactorParams& |
量化因子参数。 关于UsrQuantizeFactorParams,请参见数据类型说明。 |
返回值
参数名 |
类型 |
描述(参数说明、取值范围等) |
---|---|---|
- |
Operator& |
对象本身 |
异常处理
无。
约束说明
无。
数据类型说明
- UsrQuantizeFactorParams结构体:
struct UsrQuantizeFactorParams { uint32_t quantize_algo{0}; uint32_t scale_type{0}; UsrQuantizeFactor quantize_param; UsrQuantizeFactor dequantize_param; UsrQuantizeFactor requantize_param; UsrQuantizeCalcFactor quantizecalc_param; };
- UsrQuantizeFactor结构体:
struct UsrQuantizeFactor { //QuantizeScaleMode scale_mode; uint32_t scale_mode{0}; std::vector<uint8_t> scale_value; int64_t scale_offset{0}; std::vector<uint8_t> offset_data_value; int64_t offset_data_offset{0}; std::vector<uint8_t> offset_weight_value; int64_t offset_weight_offset{0}; std::vector<uint8_t> offset_pad_value; int64_t offset_pad_offset{0}; };
- UsrQuantizeCalcFactor结构体:
struct UsrQuantizeCalcFactor { std::vector<uint8_t> offsetw; int64_t offsetw_offset{0}; std::vector<uint8_t> offsetd; int64_t offsetd_offset{0}; std::vector<uint8_t> scalereq; int64_t scaledreq_offset{0}; std::vector<uint8_t> offsetdnext; int64_t offsetdnext_offset{0}; };