Searched defs:num_replica_ids (Results 1 – 2 of 2) sorted by relevance
118 int num_replica_ids() const { return global_.num_replica_ids(); } in num_replica_ids() function158 int num_replica_ids() const { return num_groups() * group_size(); } in num_replica_ids() function
131 int num_replica_ids = num_groups * group_size; in FromMLIR() local