jax.experimental.pallas.mosaic_gpu.layout_cast

jax.experimental.pallas.mosaic_gpu.layout_cast(x, new_layout)

Casts the layout of the given array.

Parameters:

• x (Any)

• new_layout (Layout | ParameterizedLayout)