Function: get_input_dims

Atlas A3 training products/Atlas A3 inference products

√

Atlas A2 training products/Atlas A2 inference products

√

Atlas training products

√

Atlas inference products

√

Atlas 200I/500 A2 inference products

√

model_desc

Int, pointer address of the data of the aclmdlDesc type.

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

index

Int, sequence number of the input whose dims is to be obtained, starting at 0.

dims

Dict, maximum supported combinations for dynamic-batch or dynamic-size scenarios based on the dictionary array. For details, see aclmdlIODims.

dims = {
        "name": xxx, #tensor name
        "dimCount":xxx, # Number of dimensions in the shape
        "dims": [xx, xx, xx]   # Shape
        }

In dynamic batch/image size scenarios, if the batch size is –1 or the width and height are –1, the tensor shape is dynamic.

For example, if the format of the input tensor is NCHW, then the tensor dims [-1, 3, 224, 224] indicates that the batch size is dynamic, and the tensor dims [1, 3, -1, -1] indicates that the image size is dynamic. The italic numbers are examples only.

• If the length of the tensor name is greater than 127, when dims["name"] is output, it converts the tensor name to the acl_modelId_${id}_input_${index}_${random character string} format and establishes a mapping between the converted name and the original name, you can call acl.mdl.get_tensor_real_name to transfer the converted name and obtain the original name. (If the original name is passed to the API, the original name is obtained.)
  NOTE:

  If the name of the converted tensor conflicts with the name of an existing tensor in the model, _${random string} is added to the end of the converted name. Otherwise, no random string is added.

• If the length of the tensor name is no longer than 127 characters, the tensor name is output when dims["name"] is output.

ret

Int, error code.

• 0 indicates success.
• Other values indicate failure.