Home
last modified time | relevance | path

Searched defs:group_sharding (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1518 GroupedSharding group_sharding = in Preprocess() local
1533 GroupedSharding group_sharding = in Preprocess() local
Dspmd_partitioner_util.cc2065 GroupedSharding group_sharding = GroupShardingOnDims(sharding, group_dims); in GetManualSubgroupSharding() local