Function: get_aipp_data_size

Applicability

Product

Supported

Atlas 350 Accelerator Card

x

Atlas A3 training product / Atlas A3 inference product

Atlas A2 training product / Atlas A2 inference product

Atlas training product

Atlas inference product

Atlas 200I/500 A2 inference product

Function Usage

Obtains the memory size required by the dynamic AIPP input data based on batch_size.

Prototype

  • C Prototype
    1
    aclError aclmdlGetAippDataSize(uint64_t batchSize, size_t *size)
    
  • Python Function
    1
    mem_size, ret = acl.mdl.get_aipp_data_size(batch_size)
    

Parameters

Parameter

Description

batch_size

Int, batch size of the original model.

Returns

Return Value

Description

mem_size

Int, memory size, in bytes.

ret

Int, error code.