Searched defs:resharded (Results 1 – 2 of 2) sorted by relevance
1817 auto resharded = operand.Reshard(UngroupSharding(grouped)); in PartitionDotGroupOnBatch() local1845 if (auto resharded = reshard_to_output_batch( in PartitionDotGroupOnBatch() local1854 if (auto resharded = reshard_to_output_batch( in PartitionDotGroupOnBatch() local
359 auto resharded = ReshardNoCache(target); in Reshard() local804 auto resharded = ExchangeHaloAndGetValidData( in ReshardAsWindowedInput() local844 auto update_cache = [&](PartitionedHlo resharded) { in Replicate()