Searched refs:device_coordinates (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | topology.py | 50 def __init__(self, serialized=None, mesh_shape=None, device_coordinates=None): argument
81 self._device_coordinates = np.asarray(device_coordinates, np.int32)
116 if len(proto.device_coordinates) != expected_coordinates_size:
122 len(proto.device_coordinates)))
124 coords = np.array(proto.device_coordinates, dtype=np.int32)
135 for task in xrange(self.device_coordinates.shape[0]):
136 for device in xrange(self.device_coordinates.shape[1]):
137 x, y, z = self.device_coordinates[task, device, :]
153 def device_coordinates(self): member in Topology
171 def task_ordinal_at_coordinates(self, device_coordinates): argument
[all …]
|
| D | topology_test.py | 32 device_coordinates=[[[0, 0, 0], [0, 0, 1]]],
42 original_topology.device_coordinates, new_topology.device_coordinates)
|
| D | device_assignment.py | 399 device_coordinates = topology.device_coordinates
410 replica_assignment.append(device_coordinates[task, device, :])
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | topology.proto | 26 repeated int32 device_coordinates = 4; field
|