aclmdlGetInputSizeByIndex

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Function Usage

Obtains the size (in bytes) of an input based on the model description.

Prototype

size_t aclmdlGetInputSizeByIndex(aclmdlDesc *modelDesc, size_t index)

Parameters

Parameter	Input/Output	Description
modelDesc	Input	Pointer to data of the aclmdlDesc type Call aclmdlCreateDesc to create data of the aclmdlDesc type in advance.
index	Input	Index of the input to obtain, indexed from 0

Parameter

Input/Output

Description

modelDesc

Input

Pointer to data of the aclmdlDesc type

Call aclmdlCreateDesc to create data of the aclmdlDesc type in advance.

index

Input

Index of the input to obtain, indexed from 0

Returns

If dynamic batch/image size is enabled, the input size of the maximum profile is returned. In other cases, the size of the specified input is returned. The unit is byte.

Restrictions

If the input has a dynamic shape and dimension size is set to –1 (indicating a dimension size greater than or equal to 1), the input size obtained by this call is 0. The user needs to allocate memory based on the memory requested.

Parent topic: aclmdlDesc