Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1503 absl::flat_hash_map<int64, int64> device_to_ref_group; in AlignGroupsWith() local
1506 device_to_ref_group[device] = g; in AlignGroupsWith()
1513 ref_g = device_to_ref_group[device]; in AlignGroupsWith()
1514 } else if (ref_g != device_to_ref_group[device]) { in AlignGroupsWith()