Home
last modified time | relevance | path

Searched refs:logical_id (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_placer.cc47 std::optional<DeviceAssignment::LogicalID> logical_id; in LogicalIdForDevice() local
51 if (logical_id.has_value()) { in LogicalIdForDevice()
56 logical_id.emplace(DeviceAssignment::LogicalID{r, c}); in LogicalIdForDevice()
60 if (logical_id.has_value()) { in LogicalIdForDevice()
61 return *logical_id; in LogicalIdForDevice()
70 TF_ASSIGN_OR_RETURN(const LogicalID logical_id, in ReplicaIdForDevice()
72 return logical_id.replica_id; in ReplicaIdForDevice()
Dcollective_ops_utils.cc245 TF_ASSIGN_OR_RETURN(const DeviceAssignment::LogicalID logical_id, in GetParticipatingDevices()
247 int current_replica_id = logical_id.replica_id; in GetParticipatingDevices()
248 int current_partition_id = logical_id.computation_id; in GetParticipatingDevices()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreplica_id_thunk.cc27 const DeviceAssignment::LogicalID logical_id, in ExecuteOnStream()
29 int id = kind() == Kind::kReplicaId ? logical_id.replica_id in ExecuteOnStream()
30 : logical_id.computation_id; in ExecuteOnStream()
Dnccl_collective_thunk.cc202 DeviceAssignment::LogicalID logical_id = in GetDeviceString() local
206 logical_id.replica_id, logical_id.computation_id, in GetDeviceString()
Djitrt_custom_calls.cc2105 StatusOr<DeviceAssignment::LogicalID> logical_id = in operator ()() local
2107 if (!logical_id.ok()) return AsError(logical_id); in operator ()()
2110 params.stream->ThenMemset32(&result_data, logical_id.value().replica_id, in operator ()()
2146 StatusOr<DeviceAssignment::LogicalID> logical_id = in operator ()() local
2148 if (!logical_id.ok()) return AsError(logical_id); in operator ()()
2151 params.stream->ThenMemset32(&result_data, logical_id.value().computation_id, in operator ()()
/external/pytorch/torch/fx/experimental/
Daccelerator_partitioner.py174 logical_id_to_device[d.logical_id] = d
201 for logical_id in partition.logical_device_ids:
202 device = logical_id_to_device[logical_id]
249 partition.logical_device_ids.append(d.logical_id)
341 total_size_of_graph, logical_device_id=device_with_max_mem.logical_id
455 partition.logical_device_ids.append(device.logical_id)
481 partition.logical_device_ids.append(device.logical_id)
561 logical_id = replicate_device.logical_id
563 partition.logical_device_ids.append(logical_id)
785 partition.logical_device_ids = [self.devices[i].logical_id]
[all …]
Dpartitioner_utils.py63 logical_id: int
/external/tensorflow/tensorflow/python/tpu/
Ddevice_assignment.py551 logical_id = rindex * devices_per_replica + index
553 task = logical_id // topology.num_tpus_per_task
554 device = logical_id % topology.num_tpus_per_task
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc823 const xla::DeviceAssignment::LogicalID logical_id = in __xla_cpu_runtime_PartitionId() local
827 std::memcpy(output_buffer, &logical_id.computation_id, 4); in __xla_cpu_runtime_PartitionId()
838 const xla::DeviceAssignment::LogicalID logical_id = in __xla_cpu_runtime_CollectivePermute() local
843 channel_id_present ? logical_id.computation_id : logical_id.replica_id; in __xla_cpu_runtime_CollectivePermute()
/external/crosvm/devices/src/irqchip/
Dapic.rs833 logical_id: BitField8, field