Home
last modified time | relevance | path

Searched refs:HloPosition (Results 1 – 18 of 18) 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()
393 HloPosition{param, {1}}, HloPosition{xla_while, {1}}, in TEST_F()
394 HloPosition{while_element_2, {}}, HloPosition{body_param, {1}}, in TEST_F()
395 HloPosition{body_element_1, {}}, HloPosition{add, {}}, in TEST_F()
396 HloPosition{body_tuple, {1}}, HloPosition{tuple, {1}}, in TEST_F()
397 HloPosition{cond_param, {1}}, HloPosition{negate_2, {}})); in TEST_F()
429 UnorderedElementsAre(HloPosition{constant1, {}}, in TEST_F()
[all …]
Dhlo_value.h40 struct HloPosition { struct
49 bool operator==(const HloPosition& other) const { argument
52 bool operator!=(const HloPosition& other) const { return !(*this == other); }
55 bool operator<(const HloPosition& other) const {
62 friend H AbslHashValue(H h, const HloPosition& pos) { in AbslHashValue() argument
67 std::ostream& operator<<(std::ostream& out, const HloPosition& position);
123 void SetPositionsAndComputeUses(absl::Span<const HloPosition> positions);
129 const HloPosition& defining_position() const { return positions_[0]; } in defining_position()
150 const std::vector<HloPosition>& positions() const { return positions_; } in positions()
174 std::vector<HloPosition> positions_;
Dhlo_value.cc44 const Shape& HloPosition::shape() const { in shape()
48 string HloPosition::ToString() const { in ToString()
54 std::ostream& operator<<(std::ostream& out, const HloPosition& position) { in operator <<()
75 positions_.push_back(HloPosition{instruction, index}); in HloValue()
101 for (const HloPosition& position : positions()) { in ToString()
153 absl::Span<const HloPosition> positions) { in SetPositionsAndComputeUses()
158 for (const HloPosition& position_a : positions) { in SetPositionsAndComputeUses()
160 for (const HloPosition& position_b : positions) { in SetPositionsAndComputeUses()
171 for (const HloPosition& position : positions_) { in SetPositionsAndComputeUses()
179 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()
Dmemory_space_assignment.h42 void add_chunk(const HloPosition& position, in add_chunk()
57 absl::Span<const std::pair<HloPosition, HeapSimulator::Chunk>> chunks() in chunks()
72 std::vector<std::pair<HloPosition, HeapSimulator::Chunk>> chunks_;
524 Allocation(HloPosition defining_position, MemorySpace memory_space, in Allocation()
547 virtual HloPosition defining_position() const { return defining_position_; } in defining_position()
575 HloPosition defining_position_;
604 HloPosition defining_position() const override { in defining_position()
737 std::vector<HloPosition> aliases;
750 AllocationValue(const HloValue* value, const HloPosition& position, in AllocationValue()
754 const HloPosition& defining_position() const { return defining_position_; } in defining_position()
[all …]
Dmemory_space_assignment_utils.cc50 for (const HloPosition& position : value->positions()) { in IsValueAllowedInAlternateMemory()
62 for (const HloPosition& position : value->positions()) { in IsValueAllowedInAlternateMemory()
Dhlo_dataflow_analysis.h115 const HloValueSet& GetValueSet(const HloPosition& position) const;
116 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()
282 for (const HloPosition& position : value.positions()) { in ComputeConditionalAliasedBuffers()
315 for (const HloPosition& position : value.positions()) { in ComputeInPlaceOperationAliasedBuffers()
516 for (const HloPosition& position : buffer.ComputePositions()) { in ToString()
Dhlo_buffer.h114 std::vector<HloPosition> ComputePositions() const;
Dmemory_space_assignment.cc755 std::vector<HloPosition> positions; in CreateAllocationValues()
756 for (const HloPosition& position : value->positions()) { in CreateAllocationValues()
765 [&](const HloPosition& pos1, const HloPosition& pos2) { in CreateAllocationValues()
774 const HloPosition& position = positions.at(i); in CreateAllocationValues()
855 HloPosition root_alias{ in FindAliases()
1439 for (const HloPosition& aliased_position : use.aliases) { in AllocateAllocationValues()
1655 for (const HloPosition& position : use.aliases) { in AliasedRequiredAssignmentForUse()
1779 auto is_position_in_alternate_memory = [&](const HloPosition& position) { in AreIntervalsReservedInAlternateMemory()
1797 for (const HloPosition& position : value->positions()) { in AreIntervalsReservedInAlternateMemory()
1953 const HloPosition& defining_position = in AllocateSegment()
[all …]
Dmemory_space_propagation.cc63 for (const HloPosition& position : value.positions()) { in Propagate()
Dhlo_dataflow_analysis_test.cc128 UnorderedElementsAre(HloPosition{constant1, {}})); in TEST_P()
130 UnorderedElementsAre(HloPosition{constant2, {}})); in TEST_P()
132 UnorderedElementsAre(HloPosition{add, {}})); in TEST_P()
183 UnorderedElementsAre(HloPosition{param0, {}}, HloPosition{tuple, {0}}, in TEST_P()
184 HloPosition{gte0, {}})); in TEST_P()
187 UnorderedElementsAre(HloPosition{param1, {}}, HloPosition{tuple, {1}}, in TEST_P()
188 HloPosition{gte1, {}})); in TEST_P()
190 UnorderedElementsAre(HloPosition{tuple, {}})); in TEST_P()
230 HloPosition{constant1, {}}, HloPosition{tuple, {0}}, in TEST_P()
231 HloPosition{nested_tuple, {0, 0}}, HloPosition{nested_tuple, {1, 0}}, in TEST_P()
[all …]
Dhlo_dataflow_analysis.cc379 const HloPosition& position) const { in GetValueSet()
383 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloPosition& position) { in GetValueSet()
1126 std::vector<std::vector<HloPosition>> value_positions( in Run()
1137 HloPosition{instruction, index}); in Run()
1167 for (const HloPosition& position : value->positions()) { in Verify()
1182 const HloPosition position{instruction, index}; in Verify()
Dhlo_ordering.cc398 for (const HloPosition& pos : a.positions()) { in LiveRangeStrictlyBefore()
421 for (const HloPosition& position : a.positions()) { in LiveRangeStrictlyBefore()
Dhlo_live_range.cc176 for (const HloPosition& position : value->positions()) { in CalculateBufferStartEndMap()
Dbuffer_assignment.cc261 for (HloPosition position : buffer.positions()) { in AddAssignment()
325 for (const HloPosition& position : value->positions()) { in GetOutputInstruction()
1068 for (const HloPosition& assigned_buffer_position : in MaybeAssignBuffer()
1383 const HloPosition& defining_position = position_and_chunk.first; in AssignPresetBuffers()
Dbuffer_assignment.h592 const HloPosition& defining_position = value->defining_position(); in DefaultColorer()
Dmemory_space_assignment_test.cc162 std::set<HloPosition> positions_in_preset_assignments; in CheckPresetAssignments()
164 HloPosition position = position_and_chunk.first; in CheckPresetAssignments()
3938 const HloPosition& position = position_and_chunk.first; in TEST_P()