Searched refs:comms_by_device_ordinal (Results 1 – 2 of 2) sorted by relevance
101 absl::flat_hash_map<int, NcclComm> comms_by_device_ordinal) in NcclClique() argument102 : comms_by_device_ordinal_(std::move(comms_by_device_ordinal)) {} in NcclClique()195 absl::flat_hash_map<int, NcclComm> comms_by_device_ordinal(raw_comms.size()); in CreateNcclClique() local201 comms_by_device_ordinal.emplace(device_ordinal, in CreateNcclClique()207 return std::make_unique<NcclClique>(std::move(comms_by_device_ordinal)); in CreateNcclClique()
93 absl::flat_hash_map<int, NcclComm> comms_by_device_ordinal);