View source on GitHub |
Describes a set of TPU devices.
tf.tpu.experimental.Topology(
serialized=None, mesh_shape=None, device_coordinates=None
)
Represents both the shape of the physical mesh, and the mapping between TensorFlow TPU devices to physical mesh coordinates.
Methods
cpu_device_name_at_coordinates
cpu_device_name_at_coordinates(
device_coordinates, job=None
)
Returns the CPU device attached to a logical core.
serialized
serialized()
Returns the serialized form of the topology.
task_ordinal_at_coordinates
task_ordinal_at_coordinates(
device_coordinates
)
Returns the TensorFlow task number attached to device_coordinates
.
Args | |
---|---|
device_coordinates
|
An integer sequence describing a device's physical coordinates in the TPU fabric. |
Returns | |
---|---|
Returns the TensorFlow task number that contains the TPU device with those physical coordinates. |
tpu_device_name_at_coordinates
tpu_device_name_at_coordinates(
device_coordinates, job=None
)
Returns the name of the TPU device assigned to a logical core.
tpu_device_ordinal_at_coordinates
tpu_device_ordinal_at_coordinates(
device_coordinates
)
Returns the TensorFlow device number at device_coordinates
.
Args | |
---|---|
device_coordinates
|
An integer sequence describing a device's physical coordinates in the TPU fabric. |
Returns | |
---|---|
Returns the TensorFlow device number within the task corresponding to attached to the device with those physical coordinates. |