Home
last modified time | relevance | path

Searched refs:HloBuffer (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis.h50 const HloBuffer& GetBufferContainingValue(const HloValue& value) const { in GetBufferContainingValue()
53 HloBuffer& GetBufferContainingValue(const HloValue& value) { in GetBufferContainingValue()
58 const HloBuffer& GetBuffer(HloBuffer::Id buffer_id) const { in GetBuffer()
61 HloBuffer& GetBuffer(HloBuffer::Id buffer_id) { in GetBuffer()
67 const HloBuffer& GetUniqueBufferAt(const HloInstruction* instruction,
69 HloBuffer& GetUniqueBufferAt(const HloInstruction* instruction,
75 std::vector<const HloBuffer*> ComputeBuffersAt(
81 const std::vector<HloBuffer>& buffers() const { return buffers_; } in buffers()
98 void MergeBuffers(const HloBuffer& to, const HloBuffer& from);
105 bool BufferLivesOut(const HloBuffer& buffer) const { in BufferLivesOut()
[all …]
Dhlo_buffer.h73 class HloBuffer {
78 static bool IdLessThan(const HloBuffer* a, const HloBuffer* b) { in IdLessThan()
83 static bool IdEqual(const HloBuffer* a, const HloBuffer* b) { in IdEqual()
87 HloBuffer(Id id, absl::Span<const HloValue* const> values) in HloBuffer() function
118 bool operator==(const HloBuffer& other) const;
119 bool operator!=(const HloBuffer& other) const { return !(*this == other); }
130 std::ostream& operator<<(std::ostream& out, const HloBuffer& buffer);
Dhlo_alias_analysis.cc381 const HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt()
383 std::vector<const HloBuffer*> buffers = ComputeBuffersAt(instruction, index); in GetUniqueBufferAt()
388 HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt()
395 std::vector<const HloBuffer*> HloAliasAnalysis::ComputeBuffersAt( in ComputeBuffersAt()
397 std::vector<const HloBuffer*> buffers; in ComputeBuffersAt()
404 absl::c_sort(buffers, HloBuffer::IdLessThan); in ComputeBuffersAt()
415 const HloBuffer* buffer = nullptr; in InstructionBuffersAreAmbiguous()
429 absl::flat_hash_set<const HloBuffer*> buffers_seen; in InstructionBuffersAreDistinct()
446 std::vector<const HloBuffer*> buffers_at_this_index; in InstructionBuffersAreDistinct()
448 const HloBuffer* buffer = &GetBufferContainingValue(*value); in InstructionBuffersAreDistinct()
[all …]
Dhlo_buffer.cc36 bool HloBuffer::operator==(const HloBuffer& other) const { in operator ==()
45 std::vector<HloPosition> HloBuffer::ComputePositions() const { in ComputePositions()
58 string HloBuffer::ToString() const { in ToString()
66 std::ostream& operator<<(std::ostream& out, const HloBuffer& buffer) { in operator <<()
Dbuffer_assignment.h382 bool HasAllocation(const HloBuffer& buffer) const;
389 const HloBuffer& hlo_buffer) const;
523 BufferAllocation* NewAllocation(const HloBuffer& buffer, int64 size);
526 void AddAssignment(BufferAllocation* allocation, const HloBuffer& buffer,
536 BufferAllocation* GetMutableAssignedAllocation(const HloBuffer& buffer);
539 int64 HloBufferSize(const HloBuffer& buffer) { in HloBufferSize()
658 absl::flat_hash_set<const HloBuffer*>* assigned_buffers,
663 const HloBuffer* hlo_buffer, bool is_thread_local,
689 bool MaybeAssignBuffer(BufferAllocation* allocation, const HloBuffer& buffer,
Dbuffer_assignment.cc100 bool HloBufferIsReadOnly(const HloBuffer& buffer) { in HloBufferIsReadOnly()
398 bool BufferAssignment::HasAllocation(const HloBuffer& buffer) const { in HasAllocation()
409 const HloBuffer& hlo_buffer) const { in GetAssignedAllocation()
414 const HloBuffer& buffer) { in GetMutableAssignedAllocation()
564 BufferAllocation* BufferAssignment::NewAllocation(const HloBuffer& buffer, in NewAllocation()
573 const HloBuffer& buffer, int64 offset, in AddAssignment()
985 const HloBuffer& hlo_buffer, in MaybeAssignBuffer()
1099 const HloBuffer* hlo_buffer, bool is_thread_local, in AssignSingleHloBuffer()
1247 std::vector<const HloBuffer*> sorted_buffers; in AssignBuffersForComputations()
1250 absl::flat_hash_set<const HloBuffer*> preset_assigned_buffers; in AssignBuffersForComputations()
[all …]
Dcopy_insertion.cc494 for (const HloBuffer& buffer : alias_analysis.buffers()) { in CopyRemover()
1095 absl::flat_hash_map<const HloBuffer*, ShapeIndex> seen; in AddSpecialCaseCopies()
1098 std::vector<const HloBuffer*> buffers_at_index = in AddSpecialCaseCopies()
1101 for (const HloBuffer* buffer : buffers_at_index) { in AddSpecialCaseCopies()
1187 for (const HloBuffer& buffer : alias_analysis->buffers()) { in RemoveUnnecessaryCopies()
Dhlo_alias_analysis_test.cc58 std::vector<HloBuffer> GetBuffersAt(const HloInstruction* instruction, in GetBuffersAt()
60 std::set<HloBuffer::Id> buffer_ids;
67 std::vector<HloBuffer> buffers;
68 for (HloBuffer::Id id : buffer_ids) {
75 std::vector<HloValue> GetValuesInBuffer(const HloBuffer& buffer) { in GetValuesInBuffer()
95 for (const HloBuffer& buffer : analysis_->buffers()) { in AnyValuesInSameBufferInterfere()
Dhlo_live_range.cc35 for (const HloBuffer& hlo_buffer : alias_analysis_.buffers()) { in NormalizeAliasedBuffers()
Dheap_simulator.cc221 absl::flat_hash_map<const HloBuffer*, const HloValue*> first_allocated_value; in RunComputation()
233 const HloBuffer* hlo_buffer = in RunComputation()
254 const HloBuffer* operand_buffer = in RunComputation()
Dmemory_space_assignment.cc97 for (const HloBuffer& buffer : alias_analysis.buffers()) { in FindCrossProgramPrefetchCandidate()
196 for (const HloBuffer* buffer : alias_analysis_->ComputeBuffersAt( in GetMemoryBoundedness()
1017 const HloBuffer& buffer = in AppendBufferInfoDebugString()
1068 const HloBuffer& buffer = in AppendAllocationInfoDebugString()
1739 for (const HloBuffer* buffer : in AddInputAndOutputRequiredAssignments()
1763 for (const HloBuffer* buffer : in AddInputAndOutputRequiredAssignments()
2943 const HloBuffer& buffer = alias_analysis->GetUniqueBufferAt( in ExportAndColorBuffers()
3322 const HloBuffer& buffer = in VerifyAndExportHeapSimulatorTrace()
Dmemory_space_assignment_test.cc242 HloBuffer& buffer = alias_analysis->GetUniqueBufferAt(instruction, index);