Searched refs:HloDomainMap (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_domain_map.cc | 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 …]
|
| D | hlo_domain_map.h | 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
|
| D | hlo_domain_remover.cc | 66 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloDomainMap> domain_map, in Run()
67 HloDomainMap::Create(computation, remover_->kind_)); in Run()
|
| D | hlo_domain_verifier.cc | 72 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloDomainMap> domain_map, in Run()
73 HloDomainMap::Create(computation, kind)); in Run()
|
| D | all_reduce_combiner.cc | 108 GroupKey(const HloInstruction* hlo, const HloDomainMap& domain_map) in GroupKey()
183 TF_ASSIGN_OR_RETURN(auto domain_map, HloDomainMap::Create(computation, "")); in CreateComputationGroups()
|
| D | hlo_cse.cc | 48 TF_ASSIGN_OR_RETURN(auto domain_map, HloDomainMap::Create(computation, "")); in CombineConstants()
|