Function: set_tensor_storage_format

Applicability

Product	Supported (√/x)
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas training products	x
Atlas inference products	√
Atlas 200I/500 A2 inference products	√

Function Usage

Call acl.set_tensor_storage_format to set the actual Format of the tensor after creating the tensor description by calling acl.create_tensor_desc. This API is mainly used in the PyTorch scenarios.

This API will be deprecated in future releases. Use acl.set_tensor_format instead.

Prototype

C Prototype

aclError aclSetTensorStorageFormat(aclTensorDesc *desc, aclFormat format)

Python Function

ret = acl.set_tensor_storage_format(desc, format)

Parameter Description

Parameter	Description
desc	Int, pointer address of the aclTensorDesc type. Call acl.create_tensor_desc to create data of the aclTensorDesc type in advance.
format	Int, format of the specified tensor description.

Parameter

Description

desc

Int, pointer address of the aclTensorDesc type.

Call acl.create_tensor_desc to create data of the aclTensorDesc type in advance.

format

Int, format of the specified tensor description.

Return Value Description

Return Value	Description
ret	Int, error code. 0: success. Other values: failure.

Parent topic: aclmdlTensorDesc