Home
last modified time | relevance | path

Searched refs:HloValueSet (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_value.h193 class HloValueSet {
195 HloValueSet() = default;
197 explicit HloValueSet(absl::Span<const HloValue* const> values) in HloValueSet() function
204 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()
Dhlo_value.cc216 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()
Dhlo_dataflow_analysis.cc125 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 …]
Dhlo_dataflow_analysis.h109 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,
Dhlo_alias_analysis.cc414 const HloValueSet& value_set = pair.second; in InstructionBuffersAreAmbiguous()
432 const HloValueSet& value_set = pair.second; in InstructionBuffersAreDistinct()
Dcopy_insertion.cc588 const HloValueSet& src_value_set = in CreateCopyMap()
1132 const HloValueSet& value_set = pair.second; in AddSpecialCaseCopies()
Dheap_simulator.cc251 const HloValueSet operand_value_set = in RunComputation()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc244 const HloValueSet& sources = this->GetRootValueSet().element(index); in CreateResultShapedBuffer()