函数: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,错误码。
|
约束说明 |
无 |
注意事项 |
无 |
参考资源 |
父主题: 模型加载与执行