Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc581 auto sharding_grouped = GroupShardingOnDims( in HandleScatter() local
Dspmd_partitioner.cc956 auto sharding_grouped = in ReshardToPartialReplicateWithAllGather() local
3086 auto sharding_grouped = GroupShardingOnDims(hlo->sharding(), group_dims); in HandleRng() local
Dspmd_partitioner_util.cc1551 auto sharding_grouped = GroupShardingOnDims(sharding, sharding_dims); in AlignShardingOnDims() local