jax.experimental.pallas.tpu.GridDimensionSemantics
==================================================

.. currentmodule:: jax.experimental.pallas.tpu

.. autoclass:: GridDimensionSemantics

   
   .. automethod:: __init__

   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~GridDimensionSemantics.PARALLEL
      ~GridDimensionSemantics.CORE_PARALLEL
      ~GridDimensionSemantics.SUBCORE_PARALLEL
      ~GridDimensionSemantics.ARBITRARY