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); } |
Parent topic: common_types