jaxlib.xla_extension.GpuDevice

Warning

This page was created from a pull request (#9655).

jaxlib.xla_extension.GpuDevice¶

class jaxlib.xla_extension.GpuDevice¶
__init__(*args, **kwargs)¶

Methods

__init__(*args, **kwargs)

live_buffers(self)

transfer_from_outfeed(self, arg0)

transfer_to_infeed(self, arg0)

Attributes

client

device_kind

device_vendor

host_id

Deprecated; please use process_index

id

Integer ID of this device.

platform

process_index

Integer index of this device's process.

task_id

Deprecated; please use process_index

previous

jaxlib.xla_extension.CpuDevice

next

jaxlib.xla_extension.TpuDevice