Searched refs:domain_map (Results 1 – 4 of 4) sorted by relevance
66 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloDomainMap> domain_map, in Run()69 for (auto& domain : domain_map->GetDomains()) { in Run()79 if (domain_map->IsDomainInstruction(operand)) { in Run()89 if (root != nullptr && domain_map->IsDomainInstruction(root)) { in Run()
31 auto domain_map = absl::WrapUnique(new HloDomainMap(std::move(domain_kind))); in Create() local32 TF_RETURN_IF_ERROR(domain_map->Populate(computation)); in Create()33 return std::move(domain_map); in Create()38 auto domain_map = absl::WrapUnique(new HloDomainMap(std::move(domain_kind))); in Create() local40 TF_RETURN_IF_ERROR(domain_map->Populate(computation)); in Create()42 return std::move(domain_map); in Create()
72 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloDomainMap> domain_map, in Run()75 for (auto& domain : domain_map->GetDomains()) { in Run()
47 TF_ASSIGN_OR_RETURN(auto domain_map, HloDomainMap::Create(computation, "")); in CombineConstants()75 domain_map->InSameDomain(it->second, instruction)) { in CombineConstants()