更新时间:2021-03-18 GMT+08:00
分享

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};
    };
分享:

    相关文档

    相关产品