jax.experimental.pallas.mosaic\_gpu.Layout
==========================================

.. currentmodule:: jax.experimental.pallas.mosaic_gpu

.. autoclass:: Layout

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~Layout.reduce
      ~Layout.to_mgpu
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~Layout.WGMMA
      ~Layout.WGMMA_8BIT
      ~Layout.WGMMA_UPCAST_2X
      ~Layout.WGMMA_UPCAST_4X
      ~Layout.WGMMA_TRANSPOSED
      ~Layout.WG_SPLAT
      ~Layout.WG_STRIDED
      ~Layout.TILED
      ~Layout.TCGEN05
      ~Layout.TCGEN05_TRANSPOSED
      ~Layout.TCGEN05_M64_COLLECTIVE
      ~Layout.TCGEN05_TMEM_NATIVE
      ~Layout.SMEM_GMEM_COPY
      ~Layout.TMA_GATHER_INDICES
      ~Layout.WGMMA_ROW
      ~Layout.WGMMA_COL
      ~Layout.TCGEN05_ROW
      ~Layout.TCGEN05_COL
      ~Layout.TCGEN05_TMEM_NATIVE_ROW