IFMR
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | × |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | × |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | × |
| Atlas 训练系列产品 | × |
功能说明
输入特征图重建的量化方法。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| data | 输入 | 特征图的张量。 | FLOAT、FLOAT16 | ND |
| data_min | 输入 | 特征图最小值的张量。 | FLOAT、FLOAT16 | ND |
| data_max | 输入 | 特征图最大值的张量。 | FLOAT、FLOAT16 | ND |
| cumsum | 输入 | 数据的累积求和的张量。 | INT32 | ND |
| min_percentile | 属性 | 最小初始化百分位数。 | FLOAT | - |
| max_percentile | 属性 | 最大初始化百分位数。 | FLOAT | - |
| search_range | 属性 | 搜索范围。 | LISTFLOAT | - |
| search_step | 属性 | 搜索步长。 | FLOAT | - |
| with_offset | 属性 | 是否使用偏移量。 | BOOL | - |
| quant_bits | 属性 | 量化位数,可选属性,默认值为8。 | INT | - |
| scale | 输出 | 最优尺度。 | FLOAT | ND |
| offset | 输出 | 最优偏移量。 | FLOAT | ND |
约束说明
无
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| 图模式 | test_geir_ifmr | 通过算子IR构图方式调用IFMR算子。 |