函数：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）。 |
|
参考资源 |