Home
last modified time | relevance | path

Searched refs:logical_core (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Ddevice_assignment.py35 for logical_core in range(core_assignment.shape[1]):
36 coordinates = core_assignment[replica, logical_core, :]
40 if logical_core not in task_and_cores_to_replicas[task_id]:
41 task_and_cores_to_replicas[task_id][logical_core] = set()
43 task_and_cores_to_replicas[task_id][logical_core].add(replica)
128 def coordinates(self, replica: int, logical_core: int) -> Tuple: # pylint:disable=g-bare-generic
130 return tuple(self.core_assignment[replica, logical_core, :])
132 def lookup_replicas(self, task_id: int, logical_core: int) -> List[int]:
146 return self._task_and_cores_to_replicas[task_id][logical_core]
150 format(task_id, logical_core))
[all …]
Dtpu_feed.py878 task_id=self._host_id, logical_core=0)[replica_index]
879 for logical_core in range(self._device_assignment.num_cores_per_replica):
884 replica=replica_id, logical_core=logical_core)
888 replica=replica_id, logical_core=logical_core)
901 replica_index, logical_core),
Dtpu.py366 for logical_core in range(device_assignment.num_cores_per_replica):
369 replica=replica_id, logical_core=logical_core, job=job_name))
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt31 …argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=No…
35 …argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, de…
39 …argspec: "args=[\'self\', \'task_id\', \'logical_core\'], varargs=None, keywords=None, defaults=No…
43 …argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, de…
47 …argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=[\…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt31 …argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=No…
35 …argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, de…
39 …argspec: "args=[\'self\', \'task_id\', \'logical_core\'], varargs=None, keywords=None, defaults=No…
43 …argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, de…
47 …argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=[\…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc382 for (int logical_core = 0; logical_core < num_cores_per_replica; in GetGeneralTPUExecutionDeviceAssignment() local
383 ++logical_core) { in GetGeneralTPUExecutionDeviceAssignment()
407 device_assignment(replica, logical_core) = device_id; in GetGeneralTPUExecutionDeviceAssignment()
408 auto& device_and_host = devices_and_hosts[replica][logical_core]; in GetGeneralTPUExecutionDeviceAssignment()
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py882 for logical_core in range(device_assignment.num_cores_per_replica):
887 logical_core=logical_core,
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc1681 for (int logical_core = 0; logical_core < num_cores_per_replica; in ParseDeviceAssignmentAttr() local
1682 ++logical_core) { in ParseDeviceAssignmentAttr()
1703 (*device_assignment)(replica, logical_core) = core_location; in ParseDeviceAssignmentAttr()