aclmdlQuerySizeFromMem
函数功能
根据内存中的模型数据获取模型执行时所需的权值内存大小、内存大小,同步接口。
约束说明
工作和权重内存为Device内存,而且需要用户申请和释放。
函数原型
aclError aclmdlQuerySizeFromMem(const void *model, size_t modelSize, size_t *workSize, size_t *weightSize)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
model |
输入 |
模型数据的指针。 |
modelSize |
输入 |
模型数据长度,单位Byte。 |
workSize |
输出 |
模型执行时所需的工作内存大小的指针,单位Byte。 |
weightSize |
输出 |
模型执行时所需权值内存大小的指针,单位Byte。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: 模型加载与执行