SetStorageFormat

Function Usage

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

An enumeration that contains multiple formats, such as NCHW and ND.

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);
}