Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1510 int64 ref_g = -1; in AlignGroupsWith() local
1512 if (ref_g == -1) { in AlignGroupsWith()
1513 ref_g = device_to_ref_group[device]; in AlignGroupsWith()
1514 } else if (ref_g != device_to_ref_group[device]) { in AlignGroupsWith()
1518 return ref_g; in AlignGroupsWith()
1523 int64 ref_g = unique_ref_dev_group(grouped_sharding.device_groups[g]); in AlignGroupsWith() local
1524 if (ref_g < 0 || (!ignore_group_order && g != ref_g)) { in AlignGroupsWith()
1530 grouped_sharding.device_groups[g], reference.device_groups[ref_g]); in AlignGroupsWith()