函数：get_aipp_data_size

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 训练系列产品	√
Atlas 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 200/300/500 推理产品	√

根据batch_size获取动态AIPP输入数据所需的内存大小。

C函数原型

aclError aclmdlGetAippDataSize(uint64_t batchSize, size_t *size)

python函数

mem_size, ret = acl.mdl.get_aipp_data_size(batch_size)

参数名	说明
batch_size	int，原始模型的batch size。

返回值	说明
mem_size	int，内存大小，单位Byte。
ret	int，错误码返回0表示成功。返回其它值表示失败。

返回值

说明

mem_size

int，内存大小，单位Byte。

ret

int，错误码

父主题： 设置动态AIPP参数