文档
注册

aclmdlQuerySize

函数功能

根据模型文件获取模型执行时所需的权值内存大小、工作内存大小。

约束说明

工作和权重内存为Device内存,而且需要用户申请和释放。

当由用户管理内存时,为确保内存不浪费,在申请工作内存、权值内存前,需要调用本接口查询模型运行时所需工作内存、权值内存的大小。

如果模型输入数据的Shape不确定,则不能调用aclmdlQuerySize接口查询内存大小,在加载模型时,就无法由用户管理内存,因此需选择由系统管理内存的模型加载接口(例如,aclmdlLoadFromFileaclmdlLoadFromMem)。

函数原型

aclError aclmdlQuerySize(const char *fileName, size_t *workSize, size_t *weightSize)

参数说明

参数名

输入/输出

说明

fileName

输入

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

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

说明:

关于如何获取om文件,请参见ATC工具使用指南

workSize

输出

模型执行时所需的工作内存大小的指针,单位Byte。

weightSize

输出

模型执行时所需权值内存大小的指针,单位Byte。

返回值说明

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

参考资源

接口调用流程及示例代码,参见模型加载

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词