Searched refs:replication_dims (Results 1 – 1 of 1) sorted by relevance
193 const HloSharding& sharding, absl::Span<const int64> replication_dims) { in GetPartitionGroupsForReplication() argument195 for (int64 i : replication_dims) { in GetPartitionGroupsForReplication()204 if (!absl::c_linear_search(replication_dims, i)) { in GetPartitionGroupsForReplication()