Home
last modified time | relevance | path

Searched refs:LogicalBuffer (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h71 BufferAllocation(Index index, int64 size, LogicalBuffer::Color color) in BufferAllocation()
146 LogicalBuffer::Color color() const { return color_; } in color()
155 const absl::flat_hash_map<const LogicalBuffer*, OffsetSize>&
209 Slice GetSlice(const LogicalBuffer& buffer) const;
253 const std::vector<const LogicalBuffer*>& PeakMemoryLogicalBuffers() const { in PeakMemoryLogicalBuffers()
278 void AddAssignment(const LogicalBuffer& buffer, int64 offset, int64 size);
307 LogicalBuffer::Color color_;
336 absl::flat_hash_map<const LogicalBuffer*, OffsetSize> assigned_buffers_;
342 std::vector<const LogicalBuffer*> peak_buffers_;
364 bool HasAllocation(const LogicalBuffer& buffer) const;
[all …]
Dbuffer_assignment.cc65 const LogicalBuffer* AddBufferToColocatedSet( in AddBufferToColocatedSet()
68 std::vector<const LogicalBuffer*>* colocated_set) { in AddBufferToColocatedSet()
228 const LogicalBuffer& buffer) const { in GetSlice()
233 void BufferAllocation::AddAssignment(const LogicalBuffer& buffer, int64 offset, in AddAssignment()
308 std::vector<const LogicalBuffer*> sorted_buffers; in ToString()
313 [](const LogicalBuffer* a, const LogicalBuffer* b) { in ToString()
316 for (const LogicalBuffer* buffer : sorted_buffers) { in ToString()
341 bool BufferAssignment::HasAllocation(const LogicalBuffer& buffer) const { in HasAllocation()
347 const LogicalBuffer& buffer) const { in GetAssignedAllocation()
353 const LogicalBuffer& buffer) { in GetMutableAssignedAllocation()
[all …]
Dhlo_memory_scheduler.h40 const LogicalBuffer::SizeFunction&,
48 const LogicalBuffer::SizeFunction& size_function,
56 const LogicalBuffer::SizeFunction& size_function,
64 const LogicalBuffer::SizeFunction& size_function,
74 const LogicalBuffer::SizeFunction& size_function,
82 HloModule* module, const LogicalBuffer::SizeFunction& size_function,
89 const LogicalBuffer::SizeFunction& size_function);
99 HloMemoryScheduler(const LogicalBuffer::SizeFunction& size_function,
107 LogicalBuffer::SizeFunction size_function_;
Dbuffer_liveness.h48 bool MayInterfere(const LogicalBuffer& a, const LogicalBuffer& b) const;
53 bool MaybeLiveOut(const LogicalBuffer& buffer) const;
74 for (LogicalBuffer::Id id = 0; in DefaultColorer()
78 buffer.set_color(LogicalBuffer::Color(0)); in DefaultColorer()
95 bool live_range_strictly_before(const LogicalBuffer& a,
96 const LogicalBuffer& b) const;
104 absl::flat_hash_set<const LogicalBuffer*> aliased_buffers_;
Dtuple_points_to_analysis.h72 using BufferSet = tensorflow::gtl::CompactPointerSet<const LogicalBuffer*>;
77 bool ContainsBufferAtIndex(const LogicalBuffer& buffer,
81 bool ContainsBuffer(const LogicalBuffer& buffer) const;
85 void AddPointedToBuffer(const LogicalBuffer& buffer, const ShapeIndex& index);
112 using BufferList = absl::InlinedVector<const LogicalBuffer*, 1>;
195 const LogicalBuffer& GetBuffer(LogicalBuffer::Id id) const;
199 StatusOr<const LogicalBuffer*> GetBufferDefinedAt(
207 const BufferAliasVector& GetBufferAliases(const LogicalBuffer& buffer) const;
210 LogicalBuffer::Id num_logical_buffers() const { in num_logical_buffers()
221 LogicalBuffer& logical_buffer(LogicalBuffer::Id id) const { in logical_buffer()
[all …]
Dlogical_buffer_analysis.h34 LogicalBuffer& GetBuffer(LogicalBuffer::Id id) const;
38 LogicalBuffer& GetBuffer(HloInstruction* instruction,
41 const std::vector<std::unique_ptr<LogicalBuffer>>& logical_buffers() const { in logical_buffers()
44 LogicalBuffer::Id num_logical_buffers() const { return next_buffer_id_; } in num_logical_buffers()
70 std::vector<std::unique_ptr<LogicalBuffer>> logical_buffers_;
86 LogicalBuffer*, Hasher>
90 LogicalBuffer::Id next_buffer_id_ = 0;
Dhlo_memory_scheduler.cc80 const LogicalBuffer::SizeFunction& size_function, in Run()
105 const LogicalBuffer::SizeFunction& size_function, in ListScheduler()
117 absl::flat_hash_set<const LogicalBuffer*> instr_uses; in ListScheduler()
125 buffer_uses_[instruction] = std::vector<const LogicalBuffer*>( in ListScheduler()
138 for (const LogicalBuffer* buffer : buffer_uses_.at(instruction)) { in ListScheduler()
145 for (const LogicalBuffer* live_out_buffer : in ListScheduler()
154 static bool IgnoreBuffer(const LogicalBuffer& buffer) { in IgnoreBuffer()
171 std::vector<const std::pair<const LogicalBuffer* const, int64>*>
313 for (const LogicalBuffer* buffer : buffer_uses_.at(best)) { in CreateSchedule()
372 const LogicalBuffer::SizeFunction& size_function_;
[all …]
Dlogical_buffer.cc26 LogicalBuffer::LogicalBuffer(HloInstruction* instruction, in LogicalBuffer() function in xla::LogicalBuffer
32 LogicalBuffer::~LogicalBuffer() {} in ~LogicalBuffer()
34 string LogicalBuffer::ToString() const { in ToString()
Dbuffer_liveness.cc56 for (const LogicalBuffer* aliased_buffer : in Analyze()
83 for (const LogicalBuffer* buffer : aliased_buffers_) { in ToString()
87 for (const LogicalBuffer* buffer : maybe_live_out_buffers_) { in ToString()
93 bool BufferLiveness::live_range_strictly_before(const LogicalBuffer& a, in live_range_strictly_before()
94 const LogicalBuffer& b) const { in live_range_strictly_before()
148 bool BufferLiveness::MayInterfere(const LogicalBuffer& a, in MayInterfere()
149 const LogicalBuffer& b) const { in MayInterfere()
166 bool BufferLiveness::MaybeLiveOut(const LogicalBuffer& buffer) const { in MaybeLiveOut()
Dlogical_buffer.h35 class LogicalBuffer : public BufferValue {
37 LogicalBuffer(HloInstruction* instruction, const ShapeIndex& index, Id id);
38 ~LogicalBuffer() override;
62 TF_DISALLOW_COPY_AND_ASSIGN(LogicalBuffer);
Dlayout_assignment.h74 BufferLayoutConstraint(const Layout& layout, const LogicalBuffer& buffer,
77 const LogicalBuffer& buffer() const { return *buffer_; } in buffer()
84 const LogicalBuffer* buffer_;
156 const Layout* BufferLayout(const LogicalBuffer& buffer) const;
158 const LogicalBuffer& buffer) const;
168 Status SetBufferLayout(const Layout& layout, const LogicalBuffer& buffer,
197 const std::set<LogicalBuffer::Id>& unconstrained_buffer_ids() const { in unconstrained_buffer_ids()
210 std::unordered_map<const LogicalBuffer*, BufferLayoutConstraint>
229 std::set<LogicalBuffer::Id> unconstrained_buffer_ids_;
Dtuple_points_to_analysis.cc59 absl::flat_hash_set<const LogicalBuffer*> all_points_to; in IsDistinct()
86 bool PointsToSet::ContainsBuffer(const LogicalBuffer& buffer) const { in ContainsBuffer()
97 bool PointsToSet::ContainsBufferAtIndex(const LogicalBuffer& buffer, in ContainsBufferAtIndex()
103 void PointsToSet::AddPointedToBuffer(const LogicalBuffer& buffer, in AddPointedToBuffer()
189 for (const LogicalBuffer* buffer : pointed_to_buffers) { in __anon1079e89b0602()
409 for (const LogicalBuffer* false_buffer : in HandleTupleSelect()
450 Status TuplePointsToAnalysis::VerifyBuffer(const LogicalBuffer& buffer) const { in VerifyBuffer()
473 const LogicalBuffer& TuplePointsToAnalysis::GetBuffer( in GetBuffer()
474 LogicalBuffer::Id id) const { in GetBuffer()
480 StatusOr<const LogicalBuffer*> TuplePointsToAnalysis::GetBufferDefinedAt( in GetBufferDefinedAt()
[all …]
Dlogical_buffer_analysis.cc78 LogicalBuffer& LogicalBufferAnalysis::GetBuffer(LogicalBuffer::Id id) const { in GetBuffer()
84 LogicalBuffer& LogicalBufferAnalysis::GetBuffer(HloInstruction* instruction, in GetBuffer()
93 absl::make_unique<LogicalBuffer>(instruction, index, next_buffer_id_)); in NewLogicalBuffer()
Dbuffer_value_containers.h35 for (const LogicalBuffer* buffer : logical_buffer_container) { in ToBufferValueCompactPointerSet()
47 for (const LogicalBuffer* buffer : logical_buffer_container) { in ToBufferValueFlatSet()
Dlayout_assignment.cc66 const LogicalBuffer& buffer, in BufferLayoutConstraint()
111 for (const LogicalBuffer* buffer : buffers) { in LayoutConstraints()
149 return absl::c_any_of(*output_buffers, [&](const LogicalBuffer* b) { in OperandBufferForwarded()
155 const LogicalBuffer& buffer, in SetBufferLayout()
322 const LogicalBuffer& buffer) const { in BufferLayout()
330 const LogicalBuffer& buffer) const { in GetBufferLayoutConstraint()
366 for (const LogicalBuffer* buffer : in ToString()
500 const LogicalBuffer* buffer, in AddMandatoryConstraints()
898 for (const LogicalBuffer* buffer : buffers) { in CheckLayouts()
1169 const LogicalBuffer& buffer, in GetArrayUsesOfBuffer()
[all …]
Dbuffer_assignment_test.cc94 [alignment](LogicalBuffer::Color) { return alignment; }, in RunBufferAssignment()
105 [alignment](LogicalBuffer::Color) { return alignment; }, in RunBufferAssignmentNoBuffersForConstants()
115 const LogicalBuffer& buffer) { in RunBufferAssignmentNoBuffersReuseForAdd()
121 [alignment](LogicalBuffer::Color) { return alignment; }, in RunBufferAssignmentNoBuffersReuseForAdd()
134 [alignment](LogicalBuffer::Color) { return alignment; }, in RunColoredBufferAssignment()
148 [alignment](LogicalBuffer::Color) { return alignment; }, in RunBufferAssignmentWithInstructionSequence()
582 for (LogicalBuffer::Id id = 0; in TEST_F()
586 buffer.set_color(LogicalBuffer::Color(color++)); in TEST_F()
641 for (LogicalBuffer::Id id = 0; in TEST_F()
650 buffer.set_color(LogicalBuffer::Color(1)); in TEST_F()
[all …]
Dhlo.proto347 // Serialization of LogicalBuffer.
370 // Assigned represents a single LogicalBuffer that is assigned to this
409 // The id of the LogicalBuffer that the event applies to.
418 // The id of the canonical LogicalBuffer that the buffer shares with. Only
435 // Alias represents a source LogicalBuffer, and the buffer location that
Dheap_simulator.h95 const LogicalBuffer::SizeFunction& size_function);
102 const LogicalBuffer::SizeFunction& size_function,
Dhlo_rematerialization.cc290 const LogicalBuffer* logical_buffer, in GetUsers()
414 const LogicalBuffer* logical_buffer, in CreateBufferFromLogicalBuffer()
517 absl::flat_hash_map<const LogicalBuffer*, BufferId> in MemoryUsageTracker()
523 for (const LogicalBuffer* logical_buffer : in MemoryUsageTracker()
533 const LogicalBuffer* source_logical_buffer = in MemoryUsageTracker()
Dtuple_points_to_analysis_test.cc64 const LogicalBuffer* const GetBuffer(const HloInstruction* instruction, in GetBuffer()
77 absl::Span<const LogicalBuffer* const> buffers) { in ExpectHasBuffers()
78 std::vector<const LogicalBuffer*> vec(buffers.begin(), buffers.end()); in ExpectHasBuffers()
108 const LogicalBuffer* buffer = in ExpectHasBufferAliases()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc31 /*index=*/-1, /*size=*/0, LogicalBuffer::Color(0));
140 std::vector<const LogicalBuffer*> worklist; in GetNoaliasMetadataForBuffer()
146 for (const LogicalBuffer* buffer : in GetNoaliasMetadataForBuffer()
166 for (const LogicalBuffer* buffer : worklist) { in GetNoaliasMetadataForBuffer()
Dbuffer_assignment_util.cc26 const LogicalBuffer* buffer = buffer_offset_pair.first; in InstrForConstantBufferAllocation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc147 TF_ASSIGN_OR_RETURN(const LogicalBuffer* call_result_buf, in AddBackendConstraintsToDnnConvCustomCall()
228 const LogicalBuffer* output_buffer; in AddBackendConstraints()
297 TF_ASSIGN_OR_RETURN(const LogicalBuffer* out_buf, in PropagateOperandConstraint()
312 TF_ASSIGN_OR_RETURN(const LogicalBuffer* out_buf, in PropagateOperandConstraint()
330 const LogicalBuffer& buf = buffer_constraint.buffer(); in PropagateBufferConstraint()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_noalias_test.cc67 [](LogicalBuffer::Color) { return /*alignment=*/1; }); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc228 const LogicalBuffer* buffer_source = sources[0]; in CreateResultShapedBuffer()

12