Searched refs:HloValueSet (Results 1 – 6 of 6) sorted by relevance
181 class HloValueSet {183 HloValueSet() = default;185 explicit HloValueSet(absl::Span<const HloValue* const> values) in HloValueSet() function192 bool AssignUnionOf(absl::Span<const HloValueSet* const> inputs);212 bool operator==(const HloValueSet& other) const {221 bool operator!=(const HloValueSet& other) const { return !(*this == other); }234 std::ostream& operator<<(std::ostream& out, const HloValueSet& hlo_value);240 class InstructionValueSet : public ShapeTree<HloValueSet> {242 InstructionValueSet(const Shape& shape) : ShapeTree<HloValueSet>(shape) {} in InstructionValueSet()
211 void HloValueSet::SortAndUniquifyValues() { in SortAndUniquifyValues()217 string HloValueSet::ToString() const { in ToString()225 bool HloValueSet::AssignUnionOf(absl::Span<const HloValueSet* const> inputs) { in AssignUnionOf()226 HloValueSet union_set; in AssignUnionOf()227 for (const HloValueSet* input : inputs) { in AssignUnionOf()240 bool HloValueSet::AddValue(const HloValue* value) { in AddValue()250 std::ostream& operator<<(std::ostream& out, const HloValueSet& value_set) { in operator <<()264 HloValueSet& value_set = pair.second; in AssignUnionOf()266 std::vector<const HloValueSet*> input_value_sets; in AssignUnionOf()285 ForEachElement([&out](const ShapeIndex& index, const HloValueSet& value_set) { in ToString()
80 const HloValueSet& value_set = GetValueSet(instruction, index); in ValueIsDefinedAt()129 const HloValueSet& value_set = index_value_set.second; in DeleteMarkedValues()157 const HloValueSet& value_set) { in ToString()166 const HloValueSet& top_level_value_set = in ToString()199 HloValueSet& value_set = pair.second; in Phi()287 const HloValueSet& HloDataflowAnalysis::GetValueSet( in GetValueSet()292 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloInstruction* instruction, in GetValueSet()297 const HloValueSet& HloDataflowAnalysis::GetValueSet( in GetValueSet()302 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloPosition& position) { in GetValueSet()324 const HloValueSet& operand_value_set = pair.second; in UpdateSendValueSet()[all …]
105 const HloValueSet& GetValueSet(const HloInstruction* instruction,107 const HloValueSet& GetValueSet(const HloPosition& position) const;108 HloValueSet& GetValueSet(const HloPosition& position);109 HloValueSet& GetValueSet(const HloInstruction* instruction,
379 const HloValueSet& value_set = pair.second; in InstructionBuffersAreAmbiguous()397 const HloValueSet& value_set = pair.second; in InstructionBuffersAreDistinct()
570 const HloValueSet& src_value_set = in CreateCopyMap()1058 const HloValueSet& value_set = pair.second; in AddSpecialCaseCopies()1261 const HloValueSet& value_set = pair.second; in AddCopiesForBufferAssignment()