函数: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,错误码。
|
约束说明 |
当由用户管理内存时,为节省内存资源,在申请工作内存、权值内存前,需要调用acl.mdl.query_size接口查询模型运行时所需工作内存、权值内存的大小。 如果模型输入数据的Shape不确定,则不能调用acl.mdl.query_size接口查询内存大小,在加载模型时,就无法由用户管理内存,此时需选择由系统管理内存的模型加载接口(例如,acl.mdl.load_from_file、acl.mdl.load_from_mem)。 |
参考资源 |
父主题: 模型加载与执行