Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreplica_id_thunk.cc27 TF_ASSIGN_OR_RETURN(GlobalDeviceId global_device_id, in ExecuteOnStream()
30 global_device_id)); in ExecuteOnStream()
Dnccl_collective_thunk.cc70 TF_ASSIGN_OR_RETURN(GlobalDeviceId global_device_id, in ExecuteOnStream()
75 GetParticipatingDevices(global_device_id, *params.device_assn, in ExecuteOnStream()
Dcollective_permute_thunk.cc243 TF_ASSIGN_OR_RETURN(GlobalDeviceId global_device_id, in ExecuteOnStream()
246 params.device_assn->ReplicaIdForDevice(global_device_id)); in ExecuteOnStream()
DBUILD77 "//tensorflow/compiler/xla/service:global_device_id",
460 "//tensorflow/compiler/xla/service:global_device_id",
504 "//tensorflow/compiler/xla/service:global_device_id",
538 "//tensorflow/compiler/xla/service:global_device_id",
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc272 GlobalDeviceId global_device_id(device_proto.global_device_id()); in BuildDistributedDevices() local
273 device_to_node[global_device_id] = node.node_id(); in BuildDistributedDevices()
283 gpu_device_ids[device_proto.local_device_ordinal()] = global_device_id; in BuildDistributedDevices()
286 device_proto.global_device_id(), std::move(local_device), in BuildDistributedDevices()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_configuration_ops.cc184 int32_t global_device_id = host_ordinal_to_global_device_id(j); in Compute() local
185 mapping[i].push_back(global_device_id); in Compute()
/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dprotocol.proto41 int32 global_device_id = 4; // Globally unique ID number. field
/external/tensorflow/tensorflow/compiler/xla/service/
DBUILD2910 ":global_device_id",
2937 ":global_device_id",
5335 ":global_device_id",
5351 ":global_device_id",
5414 name = "global_device_id",
5415 srcs = ["global_device_id.cc"],
5416 hdrs = ["global_device_id.h"],