aclmdlBundleGetSize
产品支持情况
|
产品 |
是否支持 |
|---|---|
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
功能说明
根据模型描述信息和图索引获取图执行时所需的权值内存大小、工作内存大小。
函数原型
aclError aclmdlBundleGetSize(const aclmdlBundleQueryInfo *queryInfo, size_t index, size_t *workSize, size_t *constWeightSize)
参数说明
|
参数名 |
输入/输出 |
说明 |
|---|---|---|
|
queryInfo |
输入 |
需提前调用aclmdlBundleCreateQueryInfo接口创建aclmdlBundleQueryInfo类型的数据。 |
|
index |
输入 |
索引。 用户调用aclmdlBundleGetQueryModelNum接口获取模型中的图总数后,这个index的取值范围:[0, (图总数-1)]。 |
|
workSize |
输出 |
图执行时所需的工作内存大小的指针,单位Byte。 此处的内存为Device内存,而且需要用户申请和释放。 |
|
weightSize |
输出 |
图执行时所需权值内存大小的指针,单位Byte。 此处的内存为Device内存,而且需要用户申请和释放。 |
返回值说明
返回0表示成功,返回其他值表示失败,请参见aclError。
父主题: 模型加载和卸载