更新时间:2021-03-18 GMT+08:00
分享

aclmdlLoadFromFileWithMem

函数功能

从文件加载离线模型数据(适配昇腾AI处理器的离线模型),由用户自行管理模型运行的内存,同步接口。

系统完成模型加载后,返回的模型ID,作为后续操作时用于识别模型的标志。

函数原型

aclError aclmdlLoadFromFileWithMem(const char *modelPath, uint32_t *modelId, void *workPtr, size_t workSize, void *weightPtr, size_t weightSize)

参数说明

参数名

输入/输出

说明

modelPath

输入

离线模型文件的存储路径,路径中包含文件名。运行程序(APP)的用户需要对该存储路径有访问权限。

此处的离线模型文件是适配昇腾AI处理器的离线模型,即*.om文件。

modelId

输出

系统完成模型加载后生成的模型ID。

workPtr

输入

Device上模型所需工作内存(存放模型输入/输出等数据)的地址,由用户自行管理。

workSize

输入

模型所需工作内存的大小,单位Byte。

weightPtr

输入

Device上模型权值内存(存放权值数据)的指针,由用户自行管理。

weightSize

输入

模型所需权值内存的大小,单位Byte。

返回值说明

返回0表示成功,返回其它值表示失败。

分享:

    相关文档

    相关产品

close