jax.experimental.pallas.tpu.BufferedRefBase
===========================================

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

.. autoclass:: BufferedRefBase

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~BufferedRefBase.__init__
      ~BufferedRefBase.advance_copy_in_slot
      ~BufferedRefBase.advance_copy_out_slot
      ~BufferedRefBase.advance_wait_in_slot
      ~BufferedRefBase.advance_wait_out_slot
      ~BufferedRefBase.bind_existing_ref
      ~BufferedRefBase.get_dma_slice
      ~BufferedRefBase.initialize_slots
      ~BufferedRefBase.unbind_refs
      ~BufferedRefBase.with_spec
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~BufferedRefBase.block_shape
      ~BufferedRefBase.buffer_type
      ~BufferedRefBase.compute_index
      ~BufferedRefBase.has_allocated_buffer
      ~BufferedRefBase.has_indirect
      ~BufferedRefBase.is_buffered
      ~BufferedRefBase.is_input
      ~BufferedRefBase.is_input_output
      ~BufferedRefBase.is_manual
      ~BufferedRefBase.is_output
      ~BufferedRefBase.is_trivial_windowing
      ~BufferedRefBase.prefetched_count
      ~BufferedRefBase.spec