Home
last modified time | relevance | path

Searched refs:HloValue (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.h92 const HloValue& GetValueDefinedAt(const HloInstruction* instruction,
94 HloValue& GetValueDefinedAt(const HloInstruction* instruction,
114 const HloValue& GetUniqueValueAt(const HloInstruction* instruction,
118 HloValue& GetUniqueValueAt(const HloInstruction* instruction,
124 const HloValue& GetValue(HloValue::Id value_id) const;
125 HloValue& GetValue(HloValue::Id value_id);
131 const std::vector<const HloValue*>& values() const { return values_vector_; } in values()
163 HloValue* NewHloValue(HloInstruction* instruction, const ShapeIndex& index,
167 void MarkValueForDeletion(HloValue::Id value_id);
230 std::unordered_map<HloValue::Id, HloValue> values_;
[all …]
Dhlo_alias_analysis.cc67 for (const HloValue* value : dataflow_.values()) { in BufferValueMap()
77 for (const HloValue* value : dataflow_.values()) { in MergeAliasedBuffers()
125 const absl::flat_hash_set<const HloValue*>& GetValuesInBuffer( in GetValuesInBuffer()
132 void NewBuffer(const HloValue& value) { in NewBuffer()
139 void MoveValueToNewBuffer(const HloValue& value) { in MoveValueToNewBuffer()
146 void MoveValueToBuffer(const HloValue& value, BufferNumber buffer_number) { in MoveValueToBuffer()
148 absl::flat_hash_set<const HloValue*>& old_value_set = in MoveValueToBuffer()
165 for (const HloValue* value : from_value_set) { in MergeBuffers()
171 BufferNumber GetBufferForValue(const HloValue& value) { in GetBufferForValue()
176 const HloValue& value, std::vector<BufferNumber>* aliased_buffers) { in ComputeInputOutputAliasedBuffers()
[all …]
Dhlo_value.h88 class HloValue : public BufferValue {
91 static bool IdLessThan(const HloValue* a, const HloValue* b) { in IdLessThan()
96 static bool IdEqual(const HloValue* a, const HloValue* b) { in IdEqual()
104 HloValue(Id id, HloInstruction* instruction, const ShapeIndex& index,
106 ~HloValue() override {} in ~HloValue()
146 bool operator==(const HloValue& other) const;
147 bool operator!=(const HloValue& other) const;
171 std::ostream& operator<<(std::ostream& out, const HloValue& hlo_value);
185 explicit HloValueSet(absl::Span<const HloValue* const> values) in HloValueSet()
196 const std::vector<const HloValue*>& values() const { return values_; } in values()
[all …]
Dhlo_value.cc70 HloValue::HloValue(HloValue::Id id, HloInstruction* instruction, in HloValue() function in xla::HloValue
77 bool HloValue::operator==(const HloValue& other) const { in operator ==()
85 bool HloValue::operator!=(const HloValue& other) const { in operator !=()
89 string HloValue::ToShortString() const { in ToShortString()
97 string HloValue::ToString(int indent) const { in ToString()
151 void HloValue::SetPositionsAndComputeUses( in SetPositionsAndComputeUses()
206 std::ostream& operator<<(std::ostream& out, const HloValue& value) { in operator <<()
212 absl::c_sort(values_, HloValue::IdLessThan); in SortAndUniquifyValues()
213 values_.erase(std::unique(values_.begin(), values_.end(), HloValue::IdEqual), in SortAndUniquifyValues()
220 absl::StrJoin(values_, ", ", [](string* result, const HloValue* value) { in ToString()
[all …]
Dhlo_ordering.h50 bool IsDefinedBefore(const HloValue& a, const HloValue& b) const;
54 bool UseIsBeforeValueDefinition(const HloUse& use, const HloValue& value,
58 bool MayInterfere(const HloValue& a, const HloValue& b,
63 bool LiveRangeStrictlyBefore(const HloValue& a, const HloValue& b,
Dhlo_dataflow_analysis.cc87 const HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt()
93 HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt()
99 HloValue* HloDataflowAnalysis::NewHloValue(HloInstruction* instruction, in NewHloValue()
113 void HloDataflowAnalysis::MarkValueForDeletion(HloValue::Id value_id) { in MarkValueForDeletion()
114 HloValue& value = values_.at(value_id); in MarkValueForDeletion()
123 absl::flat_hash_set<HloValue::Id> id_set(value_ids_to_delete_.begin(), in DeleteMarkedValues()
130 for (const HloValue* value : value_set.values()) { in DeleteMarkedValues()
141 for (HloValue::Id value_id : value_ids_to_delete_) { in DeleteMarkedValues()
159 for (const HloValue* value : value_set.values()) { in ToString()
168 for (const HloValue* value : top_level_value_set.values()) { in ToString()
[all …]
Dhlo_buffer.h87 HloBuffer(Id id, absl::Span<const HloValue* const> values) in HloBuffer()
94 const std::vector<const HloValue*>& values() const { return values_; } in values()
98 const HloValue& GetUniqueValue() const { in GetUniqueValue()
116 const std::vector<const HloValue*> values_;
Dcopy_insertion.cc44 bool IsReadonlyEntryParameterValue(const HloValue& value) { in IsReadonlyEntryParameterValue()
52 bool IsConstantValue(const HloValue& value) { in IsConstantValue()
56 bool ValueIsReadOnly(const HloValue& value) { in ValueIsReadOnly()
82 bool ShouldCopyRootValue(const HloValue& value, in ShouldCopyRootValue()
460 explicit ValueNode(const HloValue* v) : value(v) {} in ValueNode()
462 const HloValue* value;
481 absl::flat_hash_map<const HloValue*, ValueNode*> value_to_node; in CopyRemover()
488 for (const HloValue* value_a : buffer.values()) { in CopyRemover()
493 for (const HloValue* value_b : buffer.values()) { in CopyRemover()
505 std::vector<const HloValue*> values = buffer.values(); in CopyRemover()
[all …]
Dhlo_ordering.cc101 bool HloOrdering::IsDefinedBefore(const HloValue& a, const HloValue& b) const { in IsDefinedBefore()
129 auto is_body_or_condition_phi = [](const HloValue& v) { in IsDefinedBefore()
171 const HloUse& use, const HloValue& value, in UseIsBeforeValueDefinition()
256 const HloValue& a, const HloValue& b, in LiveRangeStrictlyBefore()
297 bool HloOrdering::MayInterfere(const HloValue& a, const HloValue& b, in MayInterfere()
Dhlo_alias_analysis.h51 const HloBuffer& GetBufferContainingValue(const HloValue& value) const { in GetBufferContainingValue()
54 HloBuffer& GetBufferContainingValue(const HloValue& value) { in GetBufferContainingValue()
114 absl::flat_hash_map<const HloValue*, HloBuffer*> value_to_buffer_;
Dhlo_buffer.cc47 for (const HloValue* value : values_) { in ComputePositions()
61 absl::StrJoin(values_, ", ", [](string* result, const HloValue* value) { in ToString()
Dbfloat16_propagation.h176 PrimitiveType ValueTypeAfterChange(const HloValue* value) const;
187 absl::flat_hash_set<const HloValue*> values_that_must_be_kept_as_f32_;
Dbfloat16_propagation.cc85 absl::flat_hash_set<const HloValue*> changed_root_buffers; in RevertIfFusionInternalBF16Changes()
90 for (const HloValue* value : in RevertIfFusionInternalBF16Changes()
111 for (const HloValue* value : in RevertIfFusionInternalBF16Changes()
215 for (const HloValue* value : value_set.values()) { in AllUsersConsumeBF16()
843 const HloValue* value) const { in ValueTypeAfterChange()
Dhlo_alias_analysis_test.cc61 for (const HloValue* value : analysis_->dataflow_analysis()
75 std::vector<HloValue> GetValuesInBuffer(const HloBuffer& buffer) { in GetValuesInBuffer()
76 std::vector<HloValue> values; in GetValuesInBuffer()
77 for (const HloValue* value : buffer.values()) { in GetValuesInBuffer()
84 const HloValue& GetValueDefinedAt(const HloInstruction* instruction, in GetValueDefinedAt()
96 for (const HloValue* value_a : buffer.values()) { in AnyValuesInSameBufferInterfere()
97 for (const HloValue* value_b : buffer.values()) { in AnyValuesInSameBufferInterfere()
Dhlo_dataflow_analysis_test.cc60 std::vector<HloValue> HloValuesAt(const HloInstruction* instruction, in HloValuesAt()
63 std::vector<HloValue> values;
64 for (const HloValue* value :
Dheap_simulator_test.cc805 absl::make_unique<HloValue>(id, const0, ShapeIndex{})); in DummyBufferValue()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc396 const HloValue& value = dataflow.GetUniqueValueAt(&param, pair.first); in FindConstrainedUses()