SetStorageShape

Description

Sets the StorageShape attribute of an aclTensor.

StorageShape indicates the actual layout of the aclTensor in the memory, that is, the actual shape format of OriginShape.

Prototype

void SetStorageShape(const op::Shape &shape)

Parameters

For details about gert::Shape, see gert::Shape in Basic Data Structures and APIs.

Returns

None.

Constraints

None.

Example

1
2
3
4
5
6
7
8

// Set StorageShape of the input to [1, 2, 3, 4, 5].
void Func(aclTensor *input) {
    gert::Shape newShape;
    for (int64_t i = 1; i <= 5; i++) {
        newShape.AppendDim(i);
    }
    input->SetStorageShape(newShape);
}