函数:get_aipp_data_size
产品支持情况
|
产品 |
是否支持 |
|---|---|
|
Atlas 350 加速卡 |
x |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
功能说明
根据batch_size获取动态AIPP输入数据所需的内存大小。
函数原型
- C函数原型
1aclError aclmdlGetAippDataSize(uint64_t batchSize, size_t *size)
- python函数
1mem_size, ret = acl.mdl.get_aipp_data_size(batch_size)
参数说明
|
参数名 |
说明 |
|---|---|
|
batch_size |
int,原始模型的batch size。 |
父主题: 设置动态AIPP参数