跳到主要内容

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 &#123;<br> size_t batchCount; /**模型中支持的batch分档数 */<br> uint64_t batch[ACL_MAX_BATCH_NUM]; /**模型中支持的具体分档 */<br>&#125; 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 &#123;<br> size_t hwCount; /**模型中支持的宽高分档数 */<br> uint64_t hw[ACL_MAX_HW_NUM][2]; /**模型中支持的具体分档,每组分档中,数组下标为0代表的是高,下标为1代表的是宽 */<br>&#125; aclmdlHW;<br>
hwCount等于0时,表示不支持设置档位信息,以模型中的档位为准。

返回值说明

返回0表示成功,返回非0表示失败。

父主题: aclmdlDesc

aclmdlGetCurOutputDims

函数功能

根据模型描述信息获取指定的模型输出tensor的实际维度信息。同步接口。

约束说明

当前仅支持通过本接口获取以下场景中的模型输出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

在线提单