函数: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)。 | 
| 参考资源 | 接口调用流程,参见模型加载。 | 
父主题: 模型加载与执行