Searched refs:group_tile_assignment (Results 1 – 2 of 2) sorted by relevance
48 const Array<int64>& group_tile_assignment, in PartialTile() argument51 CHECK_EQ(group_tile_assignment.num_elements(), replication_groups.size()); in PartialTile()55 auto new_tile_dims = group_tile_assignment.dimensions(); in PartialTile()61 int64 group = group_tile_assignment(group_index); in PartialTile()
71 const Array<int64>& group_tile_assignment,