Searched refs:replica_id_map (Results 1 – 1 of 1) sorted by relevance
338 absl::flat_hash_map<int, int> replica_id_map; in RunCollectiveOp() local348 replica_id_map[p.replica_id] = pos; in RunCollectiveOp()368 int participant_num = xla::FindOrDie(replica_id_map, replica_id); in RunCollectiveOp()