Searched refs:CheckReplicaGroups (Results 1 – 1 of 1) sorted by relevance
195 static Status CheckReplicaGroups(HloInstruction* hlo, in CheckReplicaGroups() function252 TF_RETURN_IF_ERROR(CheckReplicaGroups(ag, ag->use_global_device_ids())); in HandleAllGather()282 TF_RETURN_IF_ERROR(CheckReplicaGroups(ar, ar->use_global_device_ids())); in HandleAllReduce()293 TF_RETURN_IF_ERROR(CheckReplicaGroups(hlo, /*use_global_device_ids=*/false)); in HandleAllToAll()