aclSetTensorFormat
Applicability
Product |
Supported |
|---|---|
√ |
|
√ |
|
√ |
|
√ |
|
x |
Description
Sets the tensor format for internal processing on the Ascend 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 Ascend 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. |
Returns
0 on success; else, failure. For details, see aclError.