Home
last modified time | relevance | path

Searched refs:local_device_ordinal (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk.cc22 int64 local_device_ordinal = stream->parent()->device_ordinal(); in GetGlobalDeviceId() local
24 TF_RET_CHECK(0 <= local_device_ordinal && in GetGlobalDeviceId()
25 local_device_ordinal < gpu_global_device_ids->size()); in GetGlobalDeviceId()
26 return (*gpu_global_device_ids)[local_device_ordinal]; in GetGlobalDeviceId()
29 return GlobalDeviceId(local_device_ordinal); in GetGlobalDeviceId()
Dnccl_utils.cc330 const RendezvousKey& rendezvous_key, int local_device_ordinal, in AcquireNcclClique() argument
340 NcclCliqueParticipantData participant(rendezvous_key, local_device_ordinal, in AcquireNcclClique()
Dnccl_utils.h151 const RendezvousKey& rendezvous_key, int local_device_ordinal,
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc276 TF_RET_CHECK(device_proto.local_device_ordinal() >= 0 && in BuildDistributedDevices()
277 device_proto.local_device_ordinal() < in BuildDistributedDevices()
279 TF_RET_CHECK(local_device_states[device_proto.local_device_ordinal()] != in BuildDistributedDevices()
282 std::move(local_device_states[device_proto.local_device_ordinal()]); in BuildDistributedDevices()
283 gpu_device_ids[device_proto.local_device_ordinal()] = global_device_id; in BuildDistributedDevices()
/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dprotocol.proto33 int32 local_device_ordinal = 1; field