Home
last modified time | relevance | path

Searched refs:global_topology (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc264 GlobalTopologyProto global_topology; in BuildDistributedDevices() local
266 distributed_client->EnumerateDevices(local_topology, &global_topology)); in BuildDistributedDevices()
270 for (const LocalTopologyProto& node : global_topology.nodes()) { in BuildDistributedDevices()
/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dclient.h110 GlobalTopologyProto* global_topology);
Dclient.cc135 GlobalTopologyProto* global_topology) { in EnumerateDevices() argument
158 response.mutable_global_topology()->Swap(global_topology); in EnumerateDevices()
Dservice.cc49 GlobalTopologyProto* global_topology) { in BuildGlobalTopology() argument
55 global_topology->add_nodes()->Swap(&local); in BuildGlobalTopology()
Dservice.h172 GlobalTopologyProto* global_topology);
Dprotocol.proto76 GlobalTopologyProto global_topology = 1; field