Searched refs:group_level_tile_dims (Results 1 – 1 of 1) sorted by relevance
1711 std::vector<int64> group_level_tile_dims(replicated->shape().rank(), 1); in PerGroupSliceFromReplicated() local1713 group_level_tile_dims[group_dims[i]] = group_dim_sizes[i]; in PerGroupSliceFromReplicated()1715 Array<int64> group_level_tile(group_level_tile_dims); in PerGroupSliceFromReplicated()