ttnn.pad_to_tile_shape

ttnn.pad_to_tile_shape(unpadded_shape: Annotated[list[int], FixedSize(4)]) list[int]

Pads the given shape to tile shape based on specified padding options.

Parameters:

unpadded_shape (List of [int]) – The original shape of the tensor to pad.

Returns:

List of [int] – The padded shape.

Note

This functionality is planned for deprecation in the future.

Example

>>> padded_shape = ttnn.pad_to_tile_shape(unpadded_shape=[1, 2, 2, 2])