Searched refs:HloValueSet (Results 1 – 8 of 8) sorted by relevance
193 class HloValueSet {195 HloValueSet() = default;197 explicit HloValueSet(absl::Span<const HloValue* const> values) in HloValueSet() function204 bool AssignUnionOf(absl::Span<const HloValueSet* const> inputs);224 bool operator==(const HloValueSet& other) const {233 bool operator!=(const HloValueSet& other) const { return !(*this == other); }246 std::ostream& operator<<(std::ostream& out, const HloValueSet& hlo_value);252 class InstructionValueSet : public ShapeTree<HloValueSet> {255 : ShapeTree<HloValueSet>(shape) {} in InstructionValueSet()
216 void HloValueSet::SortAndUniquifyValues() { in SortAndUniquifyValues()222 string HloValueSet::ToString() const { in ToString()230 bool HloValueSet::AssignUnionOf(absl::Span<const HloValueSet* const> inputs) { in AssignUnionOf()231 HloValueSet union_set; in AssignUnionOf()232 for (const HloValueSet* input : inputs) { in AssignUnionOf()245 bool HloValueSet::AddValue(const HloValue* value) { in AddValue()255 std::ostream& operator<<(std::ostream& out, const HloValueSet& value_set) { in operator <<()277 HloValueSet& value_set = pair.second; in AssignUnionOf()279 std::vector<const HloValueSet*> input_value_sets; in AssignUnionOf()298 ForEachElement([&out](const ShapeIndex& index, const HloValueSet& value_set) { in ToString()
125 const HloValueSet& value_set = GetValueSet(instruction, index); in ValueIsDefinedAt()175 const HloValueSet& value_set = index_value_set.second; in DeleteMarkedValues()203 const HloValueSet& value_set) { in ToString()212 const HloValueSet& top_level_value_set = in ToString()257 HloValueSet& value_set = pair.second; in Phi()351 HloValueSet HloDataflowAnalysis::GetFlattenedValueSet( in GetFlattenedValueSet()353 HloValueSet value_set; in GetFlattenedValueSet()358 std::vector<const HloValueSet*> all_sets; in GetFlattenedValueSet()360 const HloValueSet& value_set = pair.second; in GetFlattenedValueSet()368 const HloValueSet& HloDataflowAnalysis::GetValueSet( in GetValueSet()[all …]
109 HloValueSet GetFlattenedValueSet(const HloInstruction* instruction) const;113 const HloValueSet& GetValueSet(const HloInstruction* instruction,115 const HloValueSet& GetValueSet(const HloPosition& position) const;116 HloValueSet& GetValueSet(const HloPosition& position);117 HloValueSet& GetValueSet(const HloInstruction* instruction,
414 const HloValueSet& value_set = pair.second; in InstructionBuffersAreAmbiguous()432 const HloValueSet& value_set = pair.second; in InstructionBuffersAreDistinct()
588 const HloValueSet& src_value_set = in CreateCopyMap()1132 const HloValueSet& value_set = pair.second; in AddSpecialCaseCopies()
251 const HloValueSet operand_value_set = in RunComputation()
244 const HloValueSet& sources = this->GetRootValueSet().element(index); in CreateResultShapedBuffer()