Home
last modified time | relevance | path

Searched refs:instruction_to_domain_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_domain_map.cc55 return FindOrDefault(instruction_to_domain_, instruction, -1); in GetDomainId()
96 int64_t domain_id = FindOrDefault(instruction_to_domain_, instruction, -1); in Populate()
150 instruction_to_domain_[instruction] = domain_id; in InsertDomain()
167 FindOrDefault(instruction_to_domain_, current_instruction, -1); in ExpandDomain()
H A Dhlo_domain_map.h124 absl::flat_hash_map<const HloInstruction*, int64_t> instruction_to_domain_; variable