SetStorageFormat

Description

Sets the storage format of an aclTensor.

StorageFormat indicates the layout format of the aclTensor in the memory, such as NCHW and ND.

Prototype

void SetStorageFormat(op::Format format)

Parameters

Parameter

Input/Output

Description

format

Input

op::Format (ge::Format) data type, an enumeration that contains multiple formats, such as NCHW and ND.

For details about ge::Format, see ge::Format in Basic Data Structures and APIs.

Returns

None.

Constraints

None.

Example

1
2
3
4
// Set the storage format of the input to ND.
void Func(const aclTensor *input) {
    input->SetStorageFormat(ge::FORMAT_ND);
}