Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1466 int64 remaining_group_index = g; 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()