Searched refs:_verify_and_get_subgroup_size (Results 1 – 1 of 1) sorted by relevance
61 def _verify_and_get_subgroup_size(self, group_assignment, num_shards): member in CrossShardOptimizer130 subgroup_size = self._verify_and_get_subgroup_size(self._group_assignment,