aclGetTensorDescByIndex

Applicability

Product

Supported

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

Atlas inference products

Atlas training products

Description

Obtains the tensor description based on the specified index of an operator.

Prototype

aclTensorDesc *aclGetTensorDescByIndex(aclTensorDesc *desc, size_t index)

Parameters

Parameter

Input/Output

Description

desc

Input

Pointer to data of the aclTensorDesc type.

Call aclmdlCreateAndGetOpDesc to obtain the tensor descriptions of all inputs/outputs of the operator.

index

Input

Sequence number of the input or output tensor to obtain, indexed starting at 0.

Must be in the range of [0, Operator input/output count – 1]. Call aclmdlCreateAndGetOpDesc to obtain the operator input or output count.

Returns

Pointer to input or output data of the aclTensorDesc type.