Searched refs:logical_core (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/python/tpu/ |
D | device_assignment.py | 35 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 …]
|
D | tpu_feed.py | 878 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),
|
D | tpu.py | 366 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/ |
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/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/compiler/mlir/tensorflow/utils/ |
D | tpu_rewrite_device_util.cc | 382 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/ |
D | tpu_strategy.py | 882 for logical_core in range(device_assignment.num_cores_per_replica): 887 logical_core=logical_core,
|
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
D | distributed_tpu_rewrite_pass.cc | 1681 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()
|