aclSetTensorFormat

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	x

Function Usage

Sets the tensor format for internal processing on the AI processor after a tensor description is created by the aclCreateTensorDesc call.

For example, if the original operator shape is 4D with format NHWC or 5D with format NDHWC, the format is automatically converted to NC1HWC0 or NDC1HWC0 during model conversion or online network build to facilitate data access efficiency on the AI processor. Along the process, the operator shape is also inferred to facilitate internal computation.

Prototype

      
           aclError aclSetTensorFormat(aclTensorDesc *desc, aclFormat format)

Parameters

Parameter	Input/Output	Description
desc	Output	Pointer to data of the aclTensorDesc type. Call aclCreateTensorDesc to create data of the aclTensorDesc type in advance.
format	Input	Format to be set. For details about the type definition, see aclFormat.

Parameter

Input/Output

Description

desc

Output

Pointer to data of the aclTensorDesc type.

Call aclCreateTensorDesc to create data of the aclTensorDesc type in advance.

format

Input

Format to be set. For details about the type definition, see aclFormat.

Returns

0 on success; else, failure. For details, see aclError.

Parent topic: aclTensorDesc