tilus.ir.RegisterLayout¶
- class tilus.ir.RegisterLayout[source]¶
Layout for register tensor.
- shape¶
The shape of the layout, which is the shape of the register tensor.
- Type:
tuple[int, …]
- mode_shape¶
The size of each mode.
- Type:
tuple[int, …]
- spatial_modes¶
The spatial modes.
- Type:
tuple[int, …]
- local_modes¶
The local modes.
- Type:
tuple[int, …]
Register Layout Operations¶
|
Create a local(...).spatial(...) layout |
|
Create a local layout in column-major order. |
|
Create a spatial layout in column-major order. |
|
Compose two layouts together. |
|
Concatenate two layouts. |
|
Divide two layouts. |
|
Flatten the layout over the dimensions between start_dim and end_dim, inclusive. |
|
Create a local layout. |
|
Permute the dimensions of the layout. |
|
Reduce the layout over the given dimensions. |
|
Create a register layout with the given shape, mode shape, spatial modes, and local modes. |
|
Reshape the layout to the given shape. |
|
Create a spatial layout. |
|
Squeeze the layout over the given dimensions. |
|
Unsqueeze the layout over the given dimensions. |