IR定义配置说明
- Excel文件配置参数说明
表1 IR原型定义参数说明 列名称
含义
是否必选
Op
算子的Operator Type
是
Classify
算子相关参数的类别,包含:- 输入:Input
- 动态输入:DYNAMIC_INPUT
- 输出:Output
- 动态输出:DYNAMIC_OUTPUT
- 属性:Attr
是
Name
算子参数的名称。
是
Type
算子参数的类型。
包含如下取值:
tensor、int、bool、float、ListInt、ListFloat等
是
TypeRange
针对类型为tensor的参数,需要配置tensor支持的类型。
包含如下取值:
fp16,fp32,double,int8,int16,int32,int64,uint8,uint16,uint32,uint64,bool等。
框架为MindSpore时,需要将tensor的类型值转换为MindSpore所支持的值:
I8_Default,I16_Default,I32_Default,I64_Default,U8_Default,U16_Default,U32_Default,U64_Default,BOOL_Default等。
否
Required
是否必须输入,有如下取值:
- TRUE
- FALSE
是
Doc
对应参数的描述。
否
Attr_Default_value
属性的默认值。
否
Format
针对类型为tensor的参数,配置为tensor支持的数据排布格式。
包含如下取值:
ND,NHWC,NCHW,HWCN,NC1HWC0,FRACTAL_Z等
否
- json IR 文件配置参数说明
创建算子工程的json IR 文件配置请参见表2。
表2 json文件配置参数说明 配置字段
类型
含义
是否必选
Op Type
-
字符串
算子的Operator Type
是
Input[xx]
-
列表
输入参数描述
否
Name
字符串
算子输入参数的名称。
是
Format
字符串或列表
针对类型为tensor的参数,配置为tensor支持的数据排布格式。如果tensor支持多个数据排布格式,请使用列表。例如["ND","NHWC"]。
包含如下取值:
ND,NHWC,NCHW,HWCN,NC1HWC0,FRACTAL_Z等。
否
Type
字符串或列表
针对类型为tensor的参数,配置为tensor支持的数据排布格式。
包含如下取值:
ND,NHWC,NCHW,HWCN,NC1HWC0,FRACTAL_Z等。
若为MindStudio框架包含如下取值:
I8_Default,I16_Default,I32_Default,I64_Default,U8_Default,U16_Default,U32_Default,U64_Default,BOOL_Default等。
是
Param Type
-
参数类型:
- Required
- Optional
- Dynamic
未配置默认为Required。
是
Output[xx]
-
列表
输出参数描述。
是
Name
字符串
算子输出参数的名称。
是
Format
字符串或列表
针对类型为tensor的参数,配置为tensor支持的数据排布格式。
包含如下取值:
ND,NHWC,NCHW,HWCN,NC1HWC0,FRACTAL_Z等。
否
Type
字符串或列表
针对类型为tensor的参数,配置为tensor支持的数据排布格式。
包含如下取值:
ND,NHWC,NCHW,HWCN,NC1HWC0,FRACTAL_Z等。
若为MindStudio框架包含如下取值:
I8_Default,I16_Default,I32_Default,I64_Default,U8_Default,U16_Default,U32_Default,U64_Default,BOOL_Default等。
是
Param Type
-
参数类型:
- Required
- Optional
- Dynamic
未配置默认为Required
是
Attribute[xx]
-
列表
属性描述
否
Name
字符串
算子属性参数的名称。
是
Type
字符串或列表
算子参数的类型。
包含如下取值:
int、bool、float、string、list_int、list_float等。
若为MindStudio框架包含如下取值:
I8_Default,I16_Default,I32_Default,I64_Default,U8_Default,U16_Default,U32_Default,U64_Default,BOOL_Default等
是
Default Value
-
默认值
否
Param Type
-
参数类型:
- Required
- Optional
未配置默认为Required。
是
IR Template可以配置多个算子。点击Add按钮增加算子
- 若存在Op Type同名算子,会以后一算子创建算子工程。
- 若Input[xx]或Output[xx]中的Name参数相同,则后一个会覆盖前一参数。
- Input[xx],Output[xx]中的Type与Format需一一对应匹配,如果没有配置Format,自动以“ND”按Type的个数一一对应补齐
示例:
Op Type
-
Reshape
Input[01]
-
-
Name
x
Format
["ND"]
Type
["float","fp16","double","int8","int16","uint8","uint32","uint16","uint64"]
Param Type
Required
Input[02]
-
-
Name
shape
Format
-
Type
["int32","int64"]
Param Type
Optional
Output[01]
-
-
Name
y
Format
["ND"]
Type
["float","fp16","double","int8","int16","uint8","uint32","uint16","uint64"]
Param Type
Dynamic
Attribute[01]
-
-
Name
axis
Type
int
Default Value
0
Param Type
Optional
Attribute[02]
-
-
Name
num_axes
Type
int
Default Value
-1
Param Type
Optional