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

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

.. autoclass:: WGMMAAccumulatorRef

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~WGMMAAccumulatorRef.__init__
      ~WGMMAAccumulatorRef.get_ref_aval
      ~WGMMAAccumulatorRef.init
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~WGMMAAccumulatorRef.shape