Searched refs:HloDomainMap (Results 1 – 5 of 5) sorted by relevance
29 /* static */ StatusOr<std::unique_ptr<HloDomainMap>> HloDomainMap::Create( in Create()31 auto domain_map = absl::WrapUnique(new HloDomainMap(std::move(domain_kind))); in Create()36 /* static */ StatusOr<std::unique_ptr<HloDomainMap>> HloDomainMap::Create( in Create()38 auto domain_map = absl::WrapUnique(new HloDomainMap(std::move(domain_kind))); in Create()45 bool HloDomainMap::InSameDomain(const HloInstruction* instruction1, in InSameDomain()52 int64 HloDomainMap::GetDomainId(const HloInstruction* instruction) const { in GetDomainId()56 int64 HloDomainMap::GetDomainMetadataId( in GetDomainMetadataId()61 Status HloDomainMap::TryProcessEmptyDomain(HloInstruction* instruction) { in TryProcessEmptyDomain()81 Status HloDomainMap::Populate(HloComputation* computation) { in Populate()107 Status HloDomainMap::PopulateDomainMetadataMap() { in PopulateDomainMetadataMap()[all …]
38 class HloDomainMap {44 static StatusOr<std::unique_ptr<HloDomainMap>> Create(51 static StatusOr<std::unique_ptr<HloDomainMap>> Create(HloModule* module,82 HloDomainMap(string domain_kind) : domain_kind_(std::move(domain_kind)) {} in HloDomainMap() function
66 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloDomainMap> domain_map, in Run()67 HloDomainMap::Create(computation, remover_->kind_)); in Run()
72 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloDomainMap> domain_map, in Run()73 HloDomainMap::Create(computation, kind)); in Run()
47 TF_ASSIGN_OR_RETURN(auto domain_map, HloDomainMap::Create(computation, "")); in CombineConstants()