Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.cc1824 int64_t ref_g = -1; in DeviceGroupsAreMatch() local
1835 int64_t ref_g = unique_ref_dev_group(rhs.device_groups[g]); in DeviceGroupsAreMatch() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.cc1369 int64_t ref_g = -1; in AlignGroupsWithInternal() local
1382 int64_t ref_g = unique_ref_dev_group(grouped_sharding.device_groups[g]); in AlignGroupsWithInternal() local