Searched refs:device_to_node (Results 1 – 1 of 1) sorted by relevance
179 absl::flat_hash_map<GlobalDeviceId, int> device_to_node) in NcclIdStore() argument182 device_to_node_(std::move(device_to_node)) {} in NcclIdStore()269 absl::flat_hash_map<GlobalDeviceId, int> device_to_node; in BuildDistributedDevices() local273 device_to_node[global_device_id] = node.node_id(); in BuildDistributedDevices()297 node_id, distributed_client, device_to_node); in BuildDistributedDevices()