Searched defs:group_assignment (Results 1 – 4 of 4) sorted by relevance
40 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
29 const std::vector<int64>& group_assignment, in Convert()
5263 DenseIntElementsAttr group_assignment; in matchAndRewrite() local