TensorDesc

Describes tensors, including the data type, data layout format, shape, and dimension of each tensor.

struct TensorDesc {
    aclDataType dtype = ACL_DT_UNDEFINED;
    aclFormat format = ACL_FORMAT_UNDEFINED;
    Dims shape;
};

Member	Data Type	Default Value	Description
dtype	aclDataType	ACL_DT_UNDEFINED	Tensor data type. For details, see aclDataType. The Atlas inference products does not support the ACL_BF16 (bf16) data type.
format	aclFormat	ACL_FORMAT_UNDEFINED	Tensor format. For details, see aclFormat.
shape	Dims	-	Tensor shape. The tensor dimension must meet the parameter constraints of Dims. For details, see Dims.

Member

Data Type

Default Value

Description

dtype

aclDataType

ACL_DT_UNDEFINED

Tensor data type.

For details, see aclDataType.

The Atlas inference products does not support the ACL_BF16 (bf16) data type.

format

aclFormat

ACL_FORMAT_UNDEFINED

Tensor format.

For details, see aclFormat.

shape

Dims

Tensor shape. The tensor dimension must meet the parameter constraints of Dims. For details, see Dims.

Parent topic: atb/types.h