Searched refs:exit_domains (Results 1 – 5 of 5) sorted by relevance
69 domain->exit_domains.insert(instruction); in TryProcessEmptyDomain()125 } else if (!domain->exit_domains.empty()) { in PopulateDomainMetadataMap()126 const HloInstruction* domain_instruction = *domain->exit_domains.begin(); in PopulateDomainMetadataMap()184 domain->exit_domains.insert(user); in ExpandDomain()
59 absl::flat_hash_set<HloInstruction*> exit_domains; member
109 for (HloInstruction* instruction : domain.exit_domains) { in VerifyDomain()
102 domain.exit_domains.contains(user)) { in LocatePassThroughDomainLinks()256 domain.exit_domains.contains(user)) { in ApplyShardingFromUsers()
1647 for (HloInstruction* d : domain.exit_domains) { in NormalizeDomain()