Searched refs:IsDomainInstruction (Results 1 – 3 of 3) sorted by relevance
66 if (IsDomainInstruction(operand)) { in TryProcessEmptyDomain()88 if (IsDomainInstruction(instruction)) { in Populate()170 if (IsDomainInstruction(operand)) { in ExpandDomain()180 if (IsDomainInstruction(user)) { in ExpandDomain()204 bool HloDomainMap::IsDomainInstruction( in IsDomainInstruction() function in xla::HloDomainMap
79 if (domain_map->IsDomainInstruction(operand)) { in Run()89 if (root != nullptr && domain_map->IsDomainInstruction(root)) { in Run()
66 bool IsDomainInstruction(const HloInstruction* instruction) const;