Searched defs:resharded (Results 1 – 2 of 2) sorted by relevance
1906 auto resharded = operand.Reshard(UngroupSharding(grouped)); in PartitionDotGroupOnBatch() local1934 if (auto resharded = reshard_to_output_batch( in PartitionDotGroupOnBatch() local1943 if (auto resharded = reshard_to_output_batch( in PartitionDotGroupOnBatch() local
363 auto resharded = ReshardNoCache(target); in Reshard() local810 auto resharded = ExchangeHaloAndGetValidData( in ReshardAsWindowedInput() local851 auto update_cache = [&](PartitionedHlo resharded) { in Replicate()