Searched refs:device_to_node (Results 1 – 2 of 2) sorted by relevance
40 absl::flat_hash_map<GlobalDeviceId, int> device_to_node) in NcclIdStore() argument43 device_to_node_(std::move(device_to_node)) {} in NcclIdStore()
361 absl::flat_hash_map<GlobalDeviceId, int> device_to_node; in BuildDistributedDevices() local365 device_to_node[global_device_id] = node.node_id(); in BuildDistributedDevices()393 node_id, distributed_client, device_to_node); in BuildDistributedDevices()