Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1547 HloSharding AlignShardingOnDims(const HloSharding& sharding, in AlignShardingOnDims() function
1823 return AlignShardingOnDims(HloSharding::PartialTile(tgt_tile_assignment), in CreateMatchingShardingOnDims()
1826 return AlignShardingOnDims(HloSharding::Tile(tgt_tile_assignment), in CreateMatchingShardingOnDims()
1910 AlignShardingOnDims(HloSharding::PartialTile(new_tile_assignment), in GatherOperandsShardedAcrossParallelDims()
1913 *to_adjust = AlignShardingOnDims(HloSharding::Tile(new_tile_assignment), in GatherOperandsShardedAcrossParallelDims()
1928 AlignShardingOnDims(new_operand_shard.ReplicateOnLastTileDim() in GatherOperandsShardedAcrossParallelDims()
Dspmd_partitioner_util.h335 HloSharding AlignShardingOnDims(const HloSharding& sharding,