Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h408 HloSharding CreateMatchingShardingOnDims(const Shape& target_shape,
Dspmd_partitioner_util.cc1795 HloSharding CreateMatchingShardingOnDims(const Shape& target_shape, in CreateMatchingShardingOnDims() function
1855 CreateMatchingShardingOnDims(indices.shape(), new_operand_shard, in GatherOperandsShardedAcrossParallelDims()
1863 CreateMatchingShardingOnDims(operand.shape(), new_index_shard, in GatherOperandsShardedAcrossParallelDims()