jax.experimental.pallas.tpu.BufferedRef
=======================================

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

.. autoclass:: BufferedRef

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~BufferedRef.__init__
      ~BufferedRef.advance_copy_in_slot
      ~BufferedRef.advance_copy_out_slot
      ~BufferedRef.advance_next_fetch
      ~BufferedRef.advance_wait_in_slot
      ~BufferedRef.advance_wait_out_slot
      ~BufferedRef.bind_existing_ref
      ~BufferedRef.compute_slice
      ~BufferedRef.copy_in
      ~BufferedRef.copy_out
      ~BufferedRef.create
      ~BufferedRef.get_dma_slice
      ~BufferedRef.initialize_slots
      ~BufferedRef.input
      ~BufferedRef.input_output
      ~BufferedRef.output
      ~BufferedRef.unbind_refs
      ~BufferedRef.wait_in
      ~BufferedRef.wait_out
      ~BufferedRef.with_next_fetch
      ~BufferedRef.with_slot_index
      ~BufferedRef.with_spec
      ~BufferedRef.with_window_ref
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~BufferedRef.block_shape
      ~BufferedRef.buffer_count
      ~BufferedRef.buffer_type
      ~BufferedRef.compute_index
      ~BufferedRef.cumulative_copy_in
      ~BufferedRef.cumulative_copy_out
      ~BufferedRef.cumulative_wait_in
      ~BufferedRef.cumulative_wait_out
      ~BufferedRef.current_copy_in_slot
      ~BufferedRef.current_copy_out_slot
      ~BufferedRef.current_ref
      ~BufferedRef.current_wait_in_slot
      ~BufferedRef.current_wait_out_slot
      ~BufferedRef.has_allocated_buffer
      ~BufferedRef.has_indirect
      ~BufferedRef.is_buffered
      ~BufferedRef.is_input
      ~BufferedRef.is_input_output
      ~BufferedRef.is_manual
      ~BufferedRef.is_output
      ~BufferedRef.is_trivial_windowing
      ~BufferedRef.next_fetch_indices
      ~BufferedRef.prefetched_count
      ~BufferedRef.spec
      ~BufferedRef.use_lookahead
      ~BufferedRef.window_ref
      ~BufferedRef.copy_in_slot
      ~BufferedRef.wait_in_slot
      ~BufferedRef.copy_out_slot
      ~BufferedRef.wait_out_slot
      ~BufferedRef.next_fetch
      ~BufferedRef.sem_recvs
      ~BufferedRef.sem_sends
      ~BufferedRef.tiling