Function: get_input_dynamic_gear_count

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
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 number of dynamic shape profiles supported by a model based on the model description.

Prototype

C Prototype

        
             aclError aclmdlGetInputDynamicGearCount(const aclmdlDesc *modelDesc, size_t index, size_t *gearCount)

Python Function

        
             gear_count, ret = acl.mdl.get_input_dynamic_gear_count(model_desc, index)

Parameters

Parameter	Description
model_desc	Int, pointer address of the aclmdlDesc model description data. Call acl.mdl.create_desc to create data of the aclmdlDesc type in advance.
index	Int, reserved. The value is fixed to –1.

Parameter

Description

model_desc

Int, pointer address of the aclmdlDesc model description data.

Call acl.mdl.create_desc to create data of the aclmdlDesc type in advance.

index

Int, reserved. The value is fixed to –1.

Returns

Return Value	Description
gear_count	Int, number of dynamic shape profiles. For example, if the input tensor of a model is 4-dimensional and the shape profiles specified by the --dynamic_dims parameter for model conversion are 1, 3, 224, 224; 2, 3, 224, 224; 1, 3, 256, 256, the number of dynamic shape profiles obtained by this API call is 3.
ret	Int, error code. 0 indicates success. Other values indicate failure.

Return Value

Description

gear_count

Int, number of dynamic shape profiles.

For example, if the input tensor of a model is 4-dimensional and the shape profiles specified by the --dynamic_dims parameter for model conversion are 1, 3, 224, 224; 2, 3, 224, 224; 1, 3, 256, 256, the number of dynamic shape profiles obtained by this API call is 3.

ret

Int, error code.

0 indicates success.
Other values indicate failure.

Restrictions

If no dynamic shape profile is set during model building, this API call returns 0 dynamic shape profiles. For details about model building, see Model Building.

Parent topic: aclmdlDesc