Home
last modified time | relevance | path

Searched refs:HloPosition (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc186 UnorderedElementsAre(HloPosition{param0, {}}, HloPosition{tuple, {0}}, in TEST_F()
187 HloPosition{gte0, {}})); in TEST_F()
212 UnorderedElementsAre(HloPosition{param0, {}}, HloPosition{tuple, {0}}, in TEST_F()
213 HloPosition{tuple, {2}})); in TEST_F()
401 HloPosition{param, {1}}, HloPosition{xla_while, {1}}, in TEST_F()
402 HloPosition{while_element_2, {}}, HloPosition{body_param, {1}}, in TEST_F()
403 HloPosition{body_element_1, {}}, HloPosition{add, {}}, in TEST_F()
404 HloPosition{body_tuple, {1}}, HloPosition{tuple, {1}}, in TEST_F()
405 HloPosition{cond_param, {1}}, HloPosition{negate_2, {}})); in TEST_F()
437 UnorderedElementsAre(HloPosition{constant1, {}}, in TEST_F()
[all …]
Dhlo_value.h39 struct HloPosition { struct
48 bool operator==(const HloPosition& other) const { argument
51 bool operator!=(const HloPosition& other) const { return !(*this == other); }
54 bool operator<(const HloPosition& other) const {
61 std::ostream& operator<<(std::ostream& out, const HloPosition& position);
111 void SetPositionsAndComputeUses(absl::Span<const HloPosition> positions);
117 const HloPosition& defining_position() const { return positions_[0]; } in defining_position()
138 const std::vector<HloPosition>& positions() const { return positions_; } in positions()
162 std::vector<HloPosition> positions_;
Dhlo_value.cc43 const Shape& HloPosition::shape() const { in shape()
47 string HloPosition::ToString() const { in ToString()
53 std::ostream& operator<<(std::ostream& out, const HloPosition& position) { in operator <<()
74 positions_.push_back(HloPosition{instruction, index}); in HloValue()
100 for (const HloPosition& position : positions()) { in ToString()
152 absl::Span<const HloPosition> positions) { in SetPositionsAndComputeUses()
157 for (const HloPosition& position_a : positions) { in SetPositionsAndComputeUses()
159 for (const HloPosition& position_b : positions) { in SetPositionsAndComputeUses()
170 for (const HloPosition& position : positions_) { in SetPositionsAndComputeUses()
178 for (const HloPosition& position : positions_) { in SetPositionsAndComputeUses()
Dhlo_buffer.cc45 std::vector<HloPosition> HloBuffer::ComputePositions() const { in ComputePositions()
46 std::vector<HloPosition> positions; in ComputePositions()
Dhlo_dataflow_analysis.h107 const HloValueSet& GetValueSet(const HloPosition& position) const;
108 HloValueSet& GetValueSet(const HloPosition& position);
Dhlo_alias_analysis.cc189 for (const HloPosition& pos : value.positions()) { in ComputeInputOutputAliasedBuffers()
246 for (const HloPosition& position : value.positions()) { in ComputeWhileAliasedBuffers()
281 for (const HloPosition& position : value.positions()) { in ComputeConditionalAliasedBuffers()
481 for (const HloPosition& position : buffer.ComputePositions()) { in ToString()
Dhlo_dataflow_analysis.cc298 const HloPosition& position) const { in GetValueSet()
302 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloPosition& position) { in GetValueSet()
856 std::vector<std::vector<HloPosition>> value_positions( in Run()
867 HloPosition{instruction, index}); in Run()
897 for (const HloPosition& position : value->positions()) { in Verify()
912 const HloPosition position{instruction, index}; in Verify()
Dhlo_buffer.h103 std::vector<HloPosition> ComputePositions() const;
Dhlo_dataflow_analysis_test.cc125 UnorderedElementsAre(HloPosition{constant1, {}})); in TEST_P()
127 UnorderedElementsAre(HloPosition{constant2, {}})); in TEST_P()
129 UnorderedElementsAre(HloPosition{add, {}})); in TEST_P()
180 UnorderedElementsAre(HloPosition{param0, {}}, HloPosition{tuple, {0}}, in TEST_P()
181 HloPosition{gte0, {}})); in TEST_P()
184 UnorderedElementsAre(HloPosition{param1, {}}, HloPosition{tuple, {1}}, in TEST_P()
185 HloPosition{gte1, {}})); in TEST_P()
187 UnorderedElementsAre(HloPosition{tuple, {}})); in TEST_P()
227 HloPosition{constant1, {}}, HloPosition{tuple, {0}}, in TEST_P()
228 HloPosition{nested_tuple, {0, 0}}, HloPosition{nested_tuple, {1, 0}}, in TEST_P()
[all …]
Dhlo_ordering.cc284 for (const HloPosition& position : a.positions()) { in LiveRangeStrictlyBefore()