Searched defs:group_assignment (Results 1 – 2 of 2) sorted by relevance
41 group_assignment=None): argument72 def _verify_and_get_subgroup_size(self, group_assignment, num_shards): argument
49 group_assignment=None, argument94 def cross_replica_sum(x, group_assignment=None, name=None): argument