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

aclmdlLoadFromMemWithMem

函数功能

从内存加载离线模型数据,由用户自行管理模型运行的内存,同步接口。

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

函数原型

aclError aclmdlLoadFromMemWithMem(const void *model, size_t modelSize, uint32_t *modelId, void *workPtr, size_t workSize, void *weightPtr, size_t weightSize)

参数说明

参数名

输入/输出

说明

model

输入

模型数据的内存地址。

应用运行在Host时,此处需申请Host上的内存;应用运行在Device时,此处需申请Device上的内存。应用的运行模式可使用aclrtGetRunMode接口获取。

modelSize

输入

模型数据长度,单位Byte。

modelId

输出

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

workPtr

输入

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

workSize

输入

模型所需工作内存的大小,单位Byte。workPtr为空时无效。

weightPtr

输入

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

weightSize

输入

模型所需权值内存的大小,单位Byte。weightPtr为空时无效。

返回值说明

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

分享:

    相关文档

    相关产品

关闭导读