aclmdlGetOutputIndexByName
aclmdlGetOutputDataType
函数功能
根据模型描述信息获取模型中指定输出的数据类型。同步接口。
函数原型
aclDataType aclmdlGetOutputDataType(const aclmdlDesc *modelDesc, size_t index)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| modelDesc | 输入 | aclmdlDesc类型的指针。 需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。 |
| index | 输入 | 指定获取第几个输出的数据类型,index值从0开始。 |
返回值说明
返回指定输出的数据类型。
父主题: aclmdlDesc
aclmdlGetInputIndexByName
函数功能
根据模型中的输入名称获取对应输入的索引编号。同步接口。
函数原型
aclError aclmdlGetInputIndexByName(const aclmdlDesc *modelDesc, const char *name, size_t *index)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| modelDesc | 输入 | aclmdlDesc类型的指针。 需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。 |
| name | 输入 | 输入名称的指针。 |
| index | 输出 | 输入的索引编号的指针。 |
返回值说明
返回0表示成功,返回非0表示失败。
父主题: aclmdlDesc
aclmdlGetOutputIndexByName
函数功能
根据模型中的输出名称获取对应输出的索引编号。同步接口。
函数原型
aclError aclmdlGetOutputIndexByName(const aclmdlDesc *modelDesc, const char *name, size_t *index)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| modelDesc | 输入 | aclmdlDesc类型的指针。 需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。 |
| name | 输入 | 输出名称的指针。 |
| index | 输出 | 输出的索引编号的指针。 |
返回值说明
返回0表示成功,返回非0表示失败。
父主题: aclmdlDesc
aclmdlGetDynamicBatch
函数功能
根据模型描述信息获取模型支持的动态Batch信息。同步接口。
函数原型
**aclError**aclmdlGetDynamicBatch(const aclmdlDesc *modelDesc, aclmdlBatch *batch)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| modelDesc | 输入 | aclmdlDesc类型的指针。 需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。 |
| batch | 输出 | text<br>const int ACL_MAX_BATCH_NUM = 128;<br>typedef struct aclmdlBatch {<br> size_t batchCount; /**模型中支持的batch分档数 */<br> uint64_t batch[ACL_MAX_BATCH_NUM]; /**模型中支持的具体分档 */<br>} aclmdlBatch;<br>batchCount等于0时,表示不支持设置档位信息,以模型中的档位为准。 |
返回值说明
返回0表示成功,返回非0表示失败。
父主题: aclmdlDesc
aclmdlGetDynamicHW
函数功能
根据模型描述信息获取模型支持的动态宽高信息。同步接口。
函数原型
**aclError**aclmdlGetDynamicHW(const aclmdlDesc *modelDesc, size_t index, aclmdlHW *hw)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| modelDesc | 输入 | aclmdlDesc类型的指针。 需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。 |
| index | 输入 | 预留参数,当前未使用,固定设置为-1。 |
| hw | 输出 | text<br>const int ACL_MAX_HW_NUM = 128;<br>typedef struct aclmdlHW {<br> size_t hwCount; /**模型中支持的宽高分档数 */<br> uint64_t hw[ACL_MAX_HW_NUM][2]; /**模型中支持的具体分档,每组分档中,数组下标为0代表的是高,下标为1代表的是宽 */<br>} aclmdlHW;<br>hwCount等于0时,表示不支持设置档位信息,以模型中的档位为准。 |
返回值说明
返回0表示成功,返回非0表示失败。
父主题: aclmdlDesc
aclmdlGetCurOutputDims
函数功能
根据模型描述信息获取指定的模型输出tensor的实际维度信息。同步接口。
约束说明
当前仅支持通过本接口获取以下场景中的模型输出tensor的维度信息:
- 通过模型转换设置多档Batch size或分辨率或维度值,实现动态Batch或动态分辨率或动态维度(ND格式)时,如果用户已调用aclmdlSetDynamicBatchSize设置Batch、或调用aclmdlSetDynamicHWSize接口设置输入图片的宽高、或调用aclmdlSetInputDynamicDims接口设置某动态维度的值,则可通过该接口获取指定模型输出tensor的实际维度信息;如果用户未调用aclmdlSetDynamicBatchSize接口、或aclmdlSetDynamicHWSize接口、或aclmdlSetInputDynamicDims接口,则通过该接口可获取最大档的维度信息。
- 固定Shape场景下,通过该接口获取指定的模型输出tensor的维度信息。
函数原型
aclError aclmdlGetCurOutputDims(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| modelDesc | 输入 | aclmdlDesc类型的指针。 需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。 |
| index | 输入 | 指定获取第几个输出的Dims,index值从0开始。 |
| dims | 输出 | 输出实际维度信息的指针。 若tensor的name长度大于127,则在输出的dims.name时,系统会将tensor的name转换为“acl_modelId_${id}output${index} ${随机字符串} ”格式(如果转换后的tensor的name与模型中已有的tensor的name冲突,则会在转换后的name尾部增加“${随机字符串} ”,否则不会增加随机字符串),并在转换后的name与原name之间建立映射关系,用户可调用aclmdlGetTensorRealName接口,传入转换后的name,获取原name(若向接口传入原name,则获取的还是原name);若tensor的name长度小于或等于127,则在输出的dims.name时,按tensor的name输出。 |
返回值说明
返回0表示成功,返回非0表示失败。
父主题: aclmdlDesc
在线提单