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)
113 if len(proto.device_coordinates) != expected_coordinates_size:
119 len(proto.device_coordinates)))
121 coords = np.array(proto.device_coordinates, dtype=np.int32)
132 for task in xrange(self.device_coordinates.shape[0]):
133 for device in xrange(self.device_coordinates.shape[1]):
134 x, y, z = self.device_coordinates[task, device, :]
150 def device_coordinates(self): member in Topology
163 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 | tpu_context.py | 298 str(device_assignment.topology.device_coordinates))
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | topology.proto | 26 repeated int32 device_coordinates = 4; field
|