jaxlib.xla\_extension.TpuDevice
===============================

.. currentmodule:: jaxlib.xla_extension

.. autoclass:: TpuDevice

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~TpuDevice.__init__
      ~TpuDevice.live_buffers
      ~TpuDevice.transfer_from_outfeed
      ~TpuDevice.transfer_to_infeed
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~TpuDevice.client
      ~TpuDevice.coords
      ~TpuDevice.core_on_chip
      ~TpuDevice.device_kind
      ~TpuDevice.host_id
      ~TpuDevice.id
      ~TpuDevice.platform
      ~TpuDevice.process_index
      ~TpuDevice.task_id