Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1469 int64 groups_in_this_dim = grouped_sharding.group_dim_sizes[i]; in UngroupSharding() local
1470 ungrouped_inds[dim] = (remaining_group_index % groups_in_this_dim) * in UngroupSharding()
1473 remaining_group_index /= groups_in_this_dim; in UngroupSharding()