Home
last modified time | relevance | path

Searched defs:sharding_grouped (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dcustom_call_handler.cc110 auto sharding_grouped = in HandleCustomCallTopK() local
Dgather_scatter_handler.cc911 auto sharding_grouped = hlo_sharding_util::GroupShardingOnDims( in PartitionScatterIndexPassthroughPartition() local
Dspmd_partitioner.cc1352 auto sharding_grouped = hlo_sharding_util::GroupShardingOnDims( in ReshardToPartialReplicateWithAllGather() local
3877 auto sharding_grouped = in HandleRng() local
Dspmd_partitioner_util.cc1434 auto sharding_grouped = in AlignShardingOnDims() local