模型信息
hbDNNGetModelNameList
获取 dnnPackedHandle 所指向模型的名称列表和个数。
- 参数
- [out]
modelNameList模型名称列表。 - [out]
modelNameCount模型名称个数。 - [in]
dnnPackedHandleHorizon DNN句柄,指向多个模型。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetModelHandle
从 dnnPackedHandle 所指向模型列表中获取一个模型的句柄。调用方可以跨函数、跨线程使用返回的 dnnHandle。
- 参数
- [out]
dnnHandleDNN句柄,指向一个模型。 - [in]
dnnPackedHandleDNN句柄,指向多个模型。 - [in]
modelName模型名称。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetInputCount
获取 dnnHandle 所指向模型输入张量的个数。
- 参数
- [out]
inputCount模型输入张量的个数。 - [in]
dnnHandleDNN句柄,指向一个模型。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetInputName
获取 dnnHandle 所指向模型输入张量的名称。
- 参数
- [out]
name模型输入张量的名称。 - [in]
dnnHandleDNN句柄,指向一个模型。 - [in]
inputIndex模型输入张量的编号。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetInputTensorProperties
获取 dnnHandle 所指向模型特定输入张量的属性。
- 参数
- [out]
properties输入张量的信息。 - [in]
dnnHandleDNN句柄,指向一个模型。 - [in]
inputIndex模型输入张量的编号。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetOutputCount
获取 dnnHandle 所指向模型输出张量的个数。
- 参数
- [out]
outputCount模型输出张量的个数。 - [in]
dnnHandleDNN句柄,指向一个模型。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetOutputName
获取 dnnHandle 所指向模型输出张量的名称。
- 参数
- [out]
name模型输出张量的名称。 - [in]
dnnHandleDNN句柄,指向一个模型。 - [in]
outputIndex模型输出张量的编号。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetOutputTensorProperties
获取 dnnHandle 所指向模型特定输出张量的属性。
- 参数
- [out]
properties输出张量的信息。 - [in]
dnnHandleDNN句柄,指向一个模型。 - [in]
outputIndex模型输出张量的编号。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetInputDesc
获取 dnnHandle 指向模型特定输入所关联的描述信息。
- 参数
- [out]
desc描述信息的首地址。 - [out]
size描述信息的大小。 - [out]
type描述信息的类型,详细类型请查看 hbDNNDescType。 - [in]
dnnHandleDNN句柄,指向一个模型。 - [in]
inputIndex模型输入的编号。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetOutputDesc
获取 dnnHandle 指向模型特定输出所关联的描述信息。
- 参数
- [out]
desc描述信息的首地址。 - [out]
size描述信息的大小。 - [out]
type描述信息的类型,详细类型请查看 hbDNNDescType。 - [in]
dnnHandleDNN句柄,指向一个模型。 - [in]
outputIndex模型输出的编号。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetModelDesc
获取 dnnHandle 指向模型所关联的描述信息。
- 参数
- [out]
desc描述信息的首地址。 - [out]
size描述信息的大小。 - [out]
type描述信息的类型,详细类型请查看 hbDNNDescType。 - [in]
dnnHandleDNN句柄,指向一个模型。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
hbDNNGetHBMDesc
获取 dnnPackedHandle 和 index 指向hbm所关联的描述信息。
- 参数
- [out]
desc描述信息的首地址。 - [out]
size描述信息的大小。 - [out]
type描述信息的类型,详细类型请查看 hbDNNDescType。 - [in]
dnnPackedHandleHorizon DNN句柄,指向多个模型。 - [in]
indexhbm索引。
- [out]
- 返回值
- 返回
0则表示API成功执行,否则执行失败。
- 返回
注解
hbDNNInitializeFromFiles 或 hbDNNInitializeFromDDR 可支持同时加载多个hbm,因此索引范围应该在[0, modelFileCount)或者[0, modelDataCount),
这里 modelFileCount 和 modelDataCount 分别为两接口入参。
