更新时间:2021-03-18 GMT+08:00
分享

aclmdlQuerySizeFromMem

函数功能

根据内存中的模型数据获取模型执行时所需的权值内存大小、内存大小,同步接口。

约束说明

执行和权重内存为Device内存,而且需要用户申请和释放

函数原型

aclError aclmdlQuerySizeFromMem(const void *model, size_t modelSize, size_t *workSize, size_t *weightSize)

参数说明

参数名

输入/输出

说明

model

输入

需要获取内存信息的模型数据。

modelSize

输入

模型数据长度,单位Byte。

workSize

输出

模型执行时所需的工作内存的大小,单位Byte。

weightSize

输出

模型执行时所需权值内存的大小,单位Byte。

返回值说明

返回0表示成功,返回其它值表示失败。

分享:

    相关文档

    相关产品

close