Function: set_tensor_storage_format

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product / Atlas A3 inference product	√
Atlas A2 training product / Atlas A2 inference product	√
Atlas training product	x
Atlas inference product	√
Atlas 200I/500 A2 inference product	√

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)

Parameters

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.

Returns

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

Parent topic: aclmdlTensorDesc