函数:query_size_from_mem
C函数原型 |
aclError aclmdlQuerySizeFromMem(const void *model, size_t modelSize, size_t *workSize, size_t *weightSize) |
---|---|
Python函数 |
work_size, weight_size, ret = acl.mdl.query_size_from_mem(model, model_size) |
函数功能 |
根据内存中的模型数据获取模型执行时所需的权值内存大小、内存大小,同步接口。 |
输入说明 |
model:int,需要获取内存信息的模型数据对应的指针地址。 model_size:int,模型数据长度,单位Byte。 |
返回值说明 |
work_size:int,模型执行时所需的工作内存的大小,单位Byte。 weight_size:int,模型执行时所需权值内存的大小,单位Byte。 ret:int,错误码。
|
约束说明 |
执行和权重内存为Device内存,而且需要用户申请和释放。 |
父主题: 模型加载与执行