Home
last modified time | relevance | path

Searched refs:partially_sharded (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc1754 auto partially_sharded = PerGroupSliceFromReplicated( in PartitionDotGroupOnBatch() local
1758 partially_sharded->set_sharding(HloSharding::Replicate()); in PartitionDotGroupOnBatch()
1759 return PartitionedHlo(partially_sharded, partially_sharded->shape(), in PartitionDotGroupOnBatch()
Dspmd_partitioner.cc460 auto partially_sharded = PerGroupSliceFromReplicated( in ReshardNoCache() local
463 partially_sharded->set_sharding(target); in ReshardNoCache()
464 return PartitionedHlo(partially_sharded, base_shape(), state_); in ReshardNoCache()