Searched refs:local_topology (Results 1 – 5 of 5) sorted by relevance
249 LocalTopologyProto local_topology; in BuildDistributedDevices() local250 local_topology.set_node_id(node_id); in BuildDistributedDevices()257 local_topology.devices_size()); in BuildDistributedDevices()258 DeviceProto* device_proto = local_topology.add_devices(); in BuildDistributedDevices()266 distributed_client->EnumerateDevices(local_topology, &global_topology)); in BuildDistributedDevices()
109 xla::Status EnumerateDevices(const LocalTopologyProto& local_topology,
134 const LocalTopologyProto& local_topology, in EnumerateDevices() argument148 *request.mutable_local_topology() = local_topology; in EnumerateDevices()
217 int node_id = request->local_topology().node_id(); in EnumerateDevices()222 local_topologies_[node_id] = request->local_topology(); in EnumerateDevices()
72 LocalTopologyProto local_topology = 3; field