SetStorageShape

Function Usage

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

Returns

None

Constraints

None

Examples

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