Skip to main content

函数:get_input_index_by_name

函数: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

函数:get_input_index_by_name

C函数原型aclError aclmdlGetInputIndexByName(const aclmdlDesc *modelDesc, const char *name, size_t *index)
Python函数index, ret = acl.mdl.get_input_index_by_name(model_desc, name)
函数功能根据模型中的输入名称获取对应输入的索引编号。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 name:str,指定输入名称。
返回值说明index:int,获取对应输入的索引编号。 ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: aclmdlDesc

函数:get_output_index_by_name

C函数原型aclError aclmdlGetOutputIndexByName(const aclmdlDesc *modelDesc, const char *name, size_t *index)
Python函数index, ret = acl.mdl.get_output_index_by_name(model_desc, name)
函数功能根据模型中的输出名称获取对应输出的索引编号。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 name:str,指定输入名称。
返回值说明index:int,获取对应输入的索引编号。 ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: aclmdlDesc

函数:get_dynamic_batch

C函数原型aclError aclmdlGetDynamicBatch(const aclmdlDesc *modelDesc, aclmdlBatch *batch)
Python函数batch, ret = acl.mdl.get_dynamic_batch(model_desc)
函数功能获取模型支持的动态Batch信息。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。
返回值说明batch:dict,获取的模型支持的动态Batch信息。
- batch = { "batchCount": int,#模型中支持的batch分档数 "batch":[xx, xx,] # 模型中支持的具体分档 } 模型中支持的最大batch数为128。
- batchCount等于0时,表示不支持设置档位信息,以模型中的档位为准。
ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: aclmdlDesc

函数:get_dynamic_hw

C函数原型aclError aclmdlGetDynamicHW(const aclmdlDesc *modelDesc, size_t index, aclmdlHW *hw)
Python函数hw_info, ret = acl.mdl.get_dynamic_hw(model_desc, index**)**
函数功能获取模型支持的动态宽高信息。同步接口。
输入说明model_desc:int,aclmdlDesc类型数据的指针地址。 index:预留参数,当前未使用,固定设置为-1。
返回值说明hw_info:dict,获取的模型支持的宽高信息
- hw_info= { "hwCount": int,#模型中支持的宽高分档数 "hw":[xx, xx,] # 模型中支持的具体分档,每组分档中,数组下标为0代表的是高,下标为1代表的是宽 } 最大的HW数量为128
- hwCount等于0时,表示不支持设置档位信息,以模型中的档位为准。
ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: aclmdlDesc

在线提单