Searched refs:replica_ranks (Results 1 – 1 of 1) sorted by relevance
355 absl::flat_hash_map<int, int> replica_ranks; in RunCollectiveOp() local358 replica_ranks[replica_id] = rank; in RunCollectiveOp()364 int rank = xla::FindOrDie(replica_ranks, sender.replica_id); in RunCollectiveOp()