下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

函数:query_size

C函数原型

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

Python函数

work_size, weight_size, ret = acl.mdl.query_size(file_name)

函数功能

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

输入说明

file_name:str,需要获取内存信息的模型路径,路径中包含文件名。运行程序(APP)的用户需要对该路径有访问权限。

返回值说明

work_size:int,模型执行时所需的工作内存的大小,单位Byte。

weight_size:int,模型执行时所需权值内存的大小,单位Byte。

ret:int,错误码。

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

约束说明

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

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

注意事项

参考资源

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

当前产品无相关内容

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