Searched defs:group_assignment (Results 1 – 2 of 2) sorted by relevance
38 group_assignment=None): argument61 def _verify_and_get_subgroup_size(self, group_assignment, num_shards): argument
48 group_assignment=None, argument92 def cross_replica_sum(x, group_assignment=None, name=None): argument