昇腾社区首页
中文
注册

函数: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 推理产品

使用说明

C函数原型

aclError aclmdlGetAippDataSize(uint64_t batchSize, size_t *size)

Python函数

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

函数功能

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

输入说明

batch_size:int,原始模型的batch size。

输出说明

mem_size:int,内存大小,单位Byte。

返回值说明

ret:int,返回0表示成功,返回非0表示失败。