昇腾社区首页
中文
注册
开发者
下载

aclmdlBundleGetVarWeightSize

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

功能说明

获取模型的可更新权重内存大小。

函数原型

aclError aclmdlBundleGetVarWeightSize(const aclmdlBundleQueryInfo *queryInfo, size_t *variableWeightSize)

参数说明

参数名

输入/输出

说明

queryInfo

输入

需提前调用aclmdlBundleCreateQueryInfo接口创建aclmdlBundleQueryInfo类型的数据。

variableWeightSize

输出

获取模型的可更新权重内存大小。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError

参考资源

接口调用流程及示例代码请参见权重更新(由用户管理内存)