Home
last modified time | relevance | path

Searched defs:resharded (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc2180 auto resharded = operand.Reshard(UngroupSharding(grouped)); in PartitionDotGroupOnBatch() local
2208 if (auto resharded = reshard_to_output_batch( in PartitionDotGroupOnBatch() local
2217 if (auto resharded = reshard_to_output_batch( in PartitionDotGroupOnBatch() local
H A Dspmd_partitioner.cc399 auto resharded = ReshardNoCache(target, std::move(pad_value)); in Reshard() local
1106 auto resharded = ExchangeHaloAndGetValidData( in ReshardAsWindowedInput() local
1153 auto update_cache = [&](PartitionedHlo resharded) { in Replicate()