Searched defs:device_coordinates (Results 1 – 2 of 2) sorted by relevance
50 def __init__(self, serialized=None, mesh_shape=None, device_coordinates=None): argument150 def device_coordinates(self): member in Topology163 def task_ordinal_at_coordinates(self, device_coordinates): argument176 def tpu_device_ordinal_at_coordinates(self, device_coordinates): argument189 def cpu_device_name_at_coordinates(self, device_coordinates, job=None): argument194 def tpu_device_name_at_coordinates(self, device_coordinates, job=None): argument
26 repeated int32 device_coordinates = 4; field