Searched refs:tpu_device_ordinal_at_coordinates (Results 1 – 4 of 4) sorted by relevance
50 name: "tpu_device_ordinal_at_coordinates"
191 def tpu_device_ordinal_at_coordinates(self, device_coordinates): member in Topology
160 return self._topology.tpu_device_ordinal_at_coordinates(coordinates)