Searched refs:ReshardNoCache (Results 1 – 2 of 2) sorted by relevance
346 PartitionedHlo ReshardNoCache(const HloSharding& target);
359 auto resharded = ReshardNoCache(target); in Reshard()369 PartitionedHlo PartitionedHlo::ReshardNoCache(const HloSharding& target) { in ReshardNoCache() function in xla::spmd::PartitionedHlo