Searched refs:device_to_node_ (Results 1 – 1 of 1) sorted by relevance
182 device_to_node_(std::move(device_to_node)) {} in NcclIdStore()189 const absl::flat_hash_map<GlobalDeviceId, int> device_to_node_; member in xla::__anon0c1acb180111::NcclIdStore208 int primary_node_id = device_to_node_.at(key.devices()[0]); in GetNcclUniqueId()