Searched refs:ungrouped_inds (Results 1 – 1 of 1) sorted by relevance
1464 std::vector<int64> ungrouped_inds(indices.begin(), indices.end()); in UngroupSharding() local1470 ungrouped_inds[dim] = (remaining_group_index % groups_in_this_dim) * in UngroupSharding()1475 tiling(ungrouped_inds) = grouped_sharding.device_groups[g][device]; in UngroupSharding()