Searched refs:target_grouped (Results 1 – 1 of 1) sorted by relevance
459 auto target_grouped = GroupShardingOnDims(target, group_dims); in ReshardNoCache() local461 hlo_, state_.partition_id, target_grouped.device_groups, group_dims, in ReshardNoCache()462 target_grouped.group_dim_sizes, state_.b); in ReshardNoCache()