Searched refs:logical_core (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | device_assignment.py | 40 for logical_core in xrange(core_assignment.shape[1]):
41 coordinates = core_assignment[replica, logical_core, :]
45 if logical_core not in task_and_cores_to_replicas[task_id]:
46 task_and_cores_to_replicas[task_id][logical_core] = set()
48 task_and_cores_to_replicas[task_id][logical_core].add(replica)
133 def coordinates(self, replica: int, logical_core: int) -> Tuple: # pylint:disable=g-bare-generic
135 return tuple(self.core_assignment[replica, logical_core, :])
137 def lookup_replicas(self, task_id: int, logical_core: int) -> List[int]:
151 return self._task_and_cores_to_replicas[task_id][logical_core]
155 format(task_id, logical_core))
[all …]
|
| D | tpu_feed.py | 877 task_id=self._host_id, logical_core=0)[replica_index]
878 for logical_core in xrange(self._device_assignment.num_cores_per_replica):
883 replica=replica_id, logical_core=logical_core)
887 replica=replica_id, logical_core=logical_core)
900 replica_index, logical_core),
|
| D | tpu.py | 348 for logical_core in range(device_assignment.num_cores_per_replica):
351 replica=replica_id, logical_core=logical_core, job=job_name))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.tpu.experimental.-device-assignment.pbtxt | 31 …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/v1/ |
| D | tensorflow.tpu.experimental.-device-assignment.pbtxt | 31 …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/ |
| D | tpu_rewrite_device_util.cc | 386 for (int logical_core = 0; logical_core < num_cores_per_replica; in GetGeneralTPUExecutionDeviceAssignment() local
387 ++logical_core) { in GetGeneralTPUExecutionDeviceAssignment()
411 device_assignment(replica, logical_core) = device_id; in GetGeneralTPUExecutionDeviceAssignment()
412 auto& device_and_host = devices_and_hosts[replica][logical_core]; in GetGeneralTPUExecutionDeviceAssignment()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 825 for logical_core in range(device_assignment.num_cores_per_replica):
830 logical_core=logical_core,
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1471 for (int logical_core = 0; logical_core < num_cores_per_replica; in ParseDeviceAssignmentAttr() local
1472 ++logical_core) { in ParseDeviceAssignmentAttr()
1493 (*device_assignment)(replica, logical_core) = core_location; in ParseDeviceAssignmentAttr()
|