Skip to main content

函数:get_input_format

函数:get_output_dims

C函数原型aclError aclmdlGetOutputDims(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims)
Python函数dims, ret = acl.mdl.get_output_dims(model_desc, index)
函数功能根据模型描述获取指定的模型输出tensor的维度信息。同步接口。 固定Shape场景下,通过该接口获取指定的模型输出tensor的维度信息。 动态Shape(动态Batch或动态分辨率)场景下,通过该接口获取最大档的维度信息。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 index:int,指定获取第几个输入的大小,index值从0开始。
返回值说明dims:dict,针对多batch或多种宽高的场景,按照该字典的数组返回其支持的最大档的组合。
text<br>dims = &#123;<br>"name": xxx, #tensor name<br>"dimCount":xxx,#shape中的维度个数<br>"dims": [xx, xx, xx] # 维度信息<br>&#125;<br>
若tensor的name长度大于127,则在输出dims.name时,pyACL会将tensor的name转换为“acl_modelId_${id}input${index}${随机字符串} ”格式(如果转换后的tensor的name与模型中已有的tensor的name冲突,则会在转换后的name尾部增加“${随机字符串} ”,否则不会增加随机字符串),并在转换后的name与原name之间建立映射关系,用户可调用acl.mdl.get_tensor_real_name接口,传入转换后的name,获取原name(若向接口传入原name,则获取的还是原name);若tensor的name长度小于或等于127,则在输出的dims.name时,按tensor的name输出。 ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: aclmdlDesc

函数:get_input_name_by_index

C函数原型const char *aclmdlGetInputNameByIndex(const aclmdlDesc *modelDesc, size_t index)
Python函数input_name = acl.mdl.get_input_name_by_index(model_desc, index)
函数功能获取模型中指定输入的输入名称。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 index:int,指定获取第几个输入的大小,index值从0开始。
返回值说明input_name:str,返回指定输入的输入名称。
约束说明

父主题: aclmdlDesc

函数:get_output_name_by_index

C函数原型const char *aclmdlGetOutputNameByIndex(const aclmdlDesc *modelDesc, size_t index)
Python函数output_name = acl.mdl.get_output_name_by_index(model_desc, index)
函数功能获取模型中指定输出的输出名称。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 index:int,指定获取第几个输入的大小,index值从0开始。
返回值说明output_name:str,返回指定输入的输入名称。 返回指定输出的输出算子名称、算子输出边的下标、top名称或输出名称。不同原始网络、不同构建模型的方式,调用本接口获取的返回值格式不同。
- Caffe网络 返回值格式如下,各项之间以冒号分割,如果模型中包含top名称就返回,不包含就不返回:
text<br> 输出算子名称 : 算子输出边下标 : top名称<br>
- TensorFlow网络
- 使用ATC工具构建om模型的场景下,返回值格式如下,各项之间以冒号分割:
text<br> 输出算子名称 : 算子输出边下标<br>
- 使用Ascend Graph接口构建om模型的场景下,返回值格式如下,各项之间以下划线分割:
text<br> output_网络输出下标_输出算子名称_算子输出边下标<br>
- ONNX网络 返回值格式如下,各项之间以冒号分割:
text<br> 输出算子名称 : 算子输出边下标 : 输出名称<br>
如果构建om模型时,不指定网络模型中输出算子的名称或网络模型中输出的名称(系统会自动从原始模型中获取输出的名称),或者仅指定网络模型中输出的名称,则返回值中会包含输出名称。 使用Ascend Graph接口构建om模型时,如果指定了网络模型中输出算子的名称,则返回值格式如下,各项之间以下划线分割:
text<br> output_网络输出下标_输出算子名称_算子输出边下标<br>
约束说明

父主题: aclmdlDesc

函数:get_input_format

C函数原型aclFormat aclmdlGetInputFormat(const aclmdlDesc *modelDesc, size_t index)
Python函数format_info = acl.mdl.get_input_format(model_desc, index)
函数功能获取模型中指定输入的Format。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 index:int,指定获取第几个输入的大小,index值从0开始。
返回值说明format_info:int,返回指定输入的Format。
- -1:ACL_FORMAT_UNDEFINED,未知格式,默认值。
- 0:ACL_FORMAT_NCHW,NCHW格式。
- 1:ACL_FORMAT_NHWC,NHWC格式。
- 2:ACL_FORMAT_ND,表示支持任意格式,仅有Square、Tanh等这些单输入对自身处理的算子外,其他需要慎用。
- 3:ACL_FORMAT_NC1HWC0,华为自研的5维数据格式。
- 4:ACL_FORMAT_FRACTAL_Z,卷积的权重格式。
约束说明

父主题: aclmdlDesc

函数:get_output_format

C函数原型aclFormat aclmdlGetOutputFormat(const aclmdlDesc *modelDesc, size_t index)
Python函数format_info = acl.mdl.get_output_format(model_desc, index)
函数功能获取模型中指定输出的Format。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 index:int,指定获取第几个输入的大小,index值从0开始。
返回值说明format_info:int,返回指定输入的Format。
- -1:ACL_FORMAT_UNDEFINED,未知格式,默认值。
- 0:ACL_FORMAT_NCHW,NCHW格式。
- 1:ACL_FORMAT_NHWC,NHWC格式。
- 2:ACL_FORMAT_ND,表示支持任意格式,仅有Square、Tanh等这些单输入对自身处理的算子外其他需要慎用。
- 3:ACL_FORMAT_NC1HWC0,华为自研的5维数据格式。
- 4:ACL_FORMAT_FRACTAL_Z,卷积的权重格式。
约束说明

父主题: aclmdlDesc

函数:get_input_data_type

C函数原型aclDataType aclmdlGetInputDataType(const aclmdlDesc *modelDesc, size_t index)
Python函数data_type = acl.mdl.get_input_data_type(model_desc, index)
函数功能获取模型中指定输入的数据类型。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 index:int,指定获取第几个输入的大小,index值从0开始。
返回值说明data_type:int,返回指定输入的数据类型。
- -1:ACL_DT_UNDEFINED,未知数据类型,默认值。
- 0:ACL_FLOAT
- 1:ACL_FLOAT16
- 2:ACL_INT8
- 3:ACL_INT32
- 4:ACL_UINT8
- 6:ACL_INT16
- 7:ACL_UINT16
- 8:ACL_UINT32
- 9:ACL_INT64
- 10:ACL_UINT64
- 11:ACL_DOUBLE
- 12:ACL_BOOL
约束说明

父主题: aclmdlDesc

函数:get_output_data_type

C函数原型aclDataType aclmdlGetOutputDataType(const aclmdlDesc *modelDesc, size_t index)
Python函数data_type = acl.mdl.get_output_data_type(model_desc, index)
函数功能根据模型描述信息获取模型中指定输出的数据类型。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 index:int,指定获取第几个输入的大小,index值从0开始。
返回值说明ata_type:int,返回指定输入的数据类型。
- -1:ACL_DT_UNDEFINED,未知数据类型,默认值。
- 0:ACL_FLOAT
- 1:ACL_FLOAT16
- 2:ACL_INT8
- 3:ACL_INT32
- 4:ACL_UINT8
- 6:ACL_INT16
- 7:ACL_UINT16
- 8:ACL_UINT32
- 9:ACL_INT64
- 10:ACL_UINT64
- 11:ACL_DOUBLE
- 12:ACL_BOOL
约束说明

父主题: aclmdlDesc

在线提单