Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.h45 const LogicalBuffer::SizeFunction&,
53 const LogicalBuffer::SizeFunction&,
67 const LogicalBuffer::SizeFunction& size_function,
77 const LogicalBuffer::SizeFunction& size_function,
87 const LogicalBuffer::SizeFunction& size_function,
100 const LogicalBuffer::SizeFunction& size_function,
108 const LogicalBuffer::SizeFunction& size_function, int64* peak_memory);
115 HloModule* module, const LogicalBuffer::SizeFunction& size_function,
123 const LogicalBuffer::SizeFunction& size_function);
133 HloMemoryScheduler(const LogicalBuffer::SizeFunction& size_function,
[all …]
Dlogical_buffer_analysis.h35 LogicalBuffer& GetBuffer(LogicalBuffer::Id id) const;
39 LogicalBuffer& GetBuffer(HloInstruction* instruction,
42 const std::vector<std::unique_ptr<LogicalBuffer>>& logical_buffers() const { in logical_buffers()
45 LogicalBuffer::Id num_logical_buffers() const { return next_buffer_id_; } in num_logical_buffers()
74 std::vector<std::unique_ptr<LogicalBuffer>> logical_buffers_;
90 LogicalBuffer*, Hasher>
94 LogicalBuffer::Id next_buffer_id_ = 0;
Dtuple_points_to_analysis.h73 using BufferSet = tensorflow::gtl::CompactPointerSet<const LogicalBuffer*>;
78 bool ContainsBufferAtIndex(const LogicalBuffer& buffer,
82 bool ContainsBuffer(const LogicalBuffer& buffer) const;
86 void AddPointedToBuffer(const LogicalBuffer& buffer, const ShapeIndex& index);
113 using BufferList = absl::InlinedVector<const LogicalBuffer*, 1>;
196 const LogicalBuffer& GetBuffer(LogicalBuffer::Id id) const;
200 StatusOr<const LogicalBuffer*> GetBufferDefinedAt(
208 const BufferAliasVector& GetBufferAliases(const LogicalBuffer& buffer) const;
211 LogicalBuffer::Id num_logical_buffers() const { in num_logical_buffers()
222 LogicalBuffer& logical_buffer(LogicalBuffer::Id id) const { in logical_buffer()
[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()
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.h76 BufferLayoutConstraint(const Layout& layout, const LogicalBuffer& buffer,
79 const LogicalBuffer& buffer() const { return *buffer_; } in buffer()
86 const LogicalBuffer* buffer_;
158 const Layout* BufferLayout(const LogicalBuffer& buffer) const;
160 const LogicalBuffer& buffer) const;
170 Status SetBufferLayout(const Layout& layout, const LogicalBuffer& buffer,
199 const std::set<LogicalBuffer::Id>& unconstrained_buffer_ids() const { in unconstrained_buffer_ids()
212 std::unordered_map<const LogicalBuffer*, BufferLayoutConstraint>
231 std::set<LogicalBuffer::Id> unconstrained_buffer_ids_;
342 virtual Layout GetUnconstrainedLayout(const LogicalBuffer& buffer) { in GetUnconstrainedLayout()
Dtuple_points_to_analysis.cc62 absl::flat_hash_set<const LogicalBuffer*> all_points_to; in IsDistinct()
89 bool PointsToSet::ContainsBuffer(const LogicalBuffer& buffer) const { in ContainsBuffer()
100 bool PointsToSet::ContainsBufferAtIndex(const LogicalBuffer& buffer, in ContainsBufferAtIndex()
106 void PointsToSet::AddPointedToBuffer(const LogicalBuffer& buffer, in AddPointedToBuffer()
192 for (const LogicalBuffer* buffer : pointed_to_buffers) { in __anonc906b6790602()
459 for (const LogicalBuffer* false_buffer : in HandleTupleSelect()
494 for (const LogicalBuffer* input_buffer : in HandleCustomCall()
530 Status TuplePointsToAnalysis::VerifyBuffer(const LogicalBuffer& buffer) const { in VerifyBuffer()
553 const LogicalBuffer& TuplePointsToAnalysis::GetBuffer( in GetBuffer()
554 LogicalBuffer::Id id) const { in GetBuffer()
[all …]
Dbuffer_assignment.h74 BufferAllocation(Index index, int64 size, LogicalBuffer::Color color) in BufferAllocation()
151 LogicalBuffer::Color color() const { return color_; } in color()
311 LogicalBuffer::Color color_;
500 LogicalBuffer::AlignmentFunction color_alignment, in BufferAssignment()
519 BufferAllocation* NewEmptyAllocation(int64 size, LogicalBuffer::Color color);
573 LogicalBuffer::AlignmentFunction color_alignment_;
615 LogicalBuffer::AlignmentFunction color_alignment,
636 LogicalBuffer::AlignmentFunction color_alignment,
685 BufferAssignment* assignment, LogicalBuffer::Color color);
694 absl::flat_hash_map<LogicalBuffer::Color,
Dlogical_buffer_analysis.cc80 LogicalBuffer& LogicalBufferAnalysis::GetBuffer(LogicalBuffer::Id id) const { in GetBuffer()
86 LogicalBuffer& LogicalBufferAnalysis::GetBuffer(HloInstruction* instruction, in GetBuffer()
95 absl::make_unique<LogicalBuffer>(instruction, index, next_buffer_id_)); in NewLogicalBuffer()
Dhlo_memory_scheduler.cc120 absl::flat_hash_set<const LogicalBuffer*> instr_uses; in ListScheduler()
128 buffer_uses_[instruction] = std::vector<const LogicalBuffer*>( in ListScheduler()
142 for (const LogicalBuffer* buffer : buffer_uses_.at(instruction)) { in ListScheduler()
149 for (const LogicalBuffer* live_out_buffer : in ListScheduler()
158 static bool IgnoreBuffer(const LogicalBuffer& buffer) { in IgnoreBuffer()
175 std::vector<const std::pair<const LogicalBuffer* const, int64>*>
325 for (const LogicalBuffer* buffer : buffer_uses_.at(best)) { in CreateSchedule()
392 absl::flat_hash_map<const HloInstruction*, std::vector<const LogicalBuffer*>>
397 absl::flat_hash_map<const LogicalBuffer*, int64> unscheduled_use_count_;
407 for (const LogicalBuffer* buffer : buffers) { in SumLogicalBufferSizes()
[all …]
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.cc71 const LogicalBuffer& buffer, in BufferLayoutConstraint()
116 for (const LogicalBuffer* buffer : buffers) { in LayoutConstraints()
154 return absl::c_any_of(*output_buffers, [&](const LogicalBuffer* b) { in OperandBufferForwarded()
160 const LogicalBuffer& buffer, in SetBufferLayout()
332 const LogicalBuffer& buffer) const { in BufferLayout()
340 const LogicalBuffer& buffer) const { in GetBufferLayoutConstraint()
376 for (const LogicalBuffer* buffer : in ToString()
514 const LogicalBuffer* buffer, in AddMandatoryConstraints()
951 for (const LogicalBuffer* buffer : buffers) { in CheckLayouts()
1227 const LogicalBuffer& buffer, in GetArrayUsesOfBuffer()
[all …]
Dheap_simulator.h117 const LogicalBuffer::SizeFunction& size_function);
124 const LogicalBuffer::SizeFunction& size_function,
131 const LogicalBuffer::SizeFunction& size_function,
Dbuffer_assignment_test.cc95 [alignment](LogicalBuffer::Color) { return alignment; }, in RunBufferAssignment()
105 [alignment](LogicalBuffer::Color) { return alignment; }, in RunBufferAssignmentNoBuffersForConstants()
117 [alignment](LogicalBuffer::Color) { return alignment; }, in RunBufferAssignmentNoBuffersReuseForAdd()
129 [alignment](LogicalBuffer::Color) { return alignment; }, in RunColoredBufferAssignment()
142 [alignment](LogicalBuffer::Color) { return alignment; }, in RunBufferAssignmentWithInstructionSequence()
153 [alignment](LogicalBuffer::Color) { return alignment; }, in RunBufferAssignmentWithPresetAssignments()
659 value.set_color(LogicalBuffer::Color(1)); in TEST_F()
663 value.set_color(LogicalBuffer::Color(0)); in TEST_F()
745 EXPECT_EQ(paramscalar_buffer.color(), LogicalBuffer::Color(0)); in TEST_F()
747 EXPECT_EQ(param0_buffer.color(), LogicalBuffer::Color(0)); in TEST_F()
[all …]
Dhlo.proto405 // Serialization of LogicalBuffer.
428 // Assigned represents a single LogicalBuffer that is assigned to this
467 // The id of the LogicalBuffer that the event applies to.
476 // The id of the canonical LogicalBuffer that the buffer shares with. Only
493 // Alias represents a source LogicalBuffer, and the buffer location that
Dbuffer_assignment.cc557 int64 size, LogicalBuffer::Color color) { in NewEmptyAllocation()
912 LogicalBuffer::AlignmentFunction color_alignment, in Run()
1346 flat_hash_map<LogicalBuffer::Color, flat_hash_set<const HloValue*>>
1349 flat_hash_map<LogicalBuffer::Color, flat_hash_set<const HloValue*>> color_map; in SplitBuffersByColor()
1364 absl::flat_hash_map<LogicalBuffer::Color, BufferAllocation*> in AssignPresetBuffers()
1367 LogicalBuffer::Color color(color_and_info.first); in AssignPresetBuffers()
1628 LogicalBuffer::AlignmentFunction color_alignment, in CreateAssignment()
Dtuple_points_to_analysis_test.cc66 const LogicalBuffer* const GetBuffer(const HloInstruction* instruction, in GetBuffer()
79 absl::Span<const LogicalBuffer* const> buffers) { in ExpectHasBuffers()
80 std::vector<const LogicalBuffer*> vec(buffers.begin(), buffers.end()); in ExpectHasBuffers()
110 const LogicalBuffer* buffer = in ExpectHasBufferAliases()
Dhlo_rematerialization.cc446 const LogicalBuffer* logical_buffer, in GetUsers()
654 const LogicalBuffer* logical_buffer, in CreateBufferFromLogicalBuffer()
796 absl::flat_hash_map<const LogicalBuffer*, BufferId> in MemoryUsageTracker()
801 for (const LogicalBuffer* logical_buffer : in MemoryUsageTracker()
811 const LogicalBuffer* source_logical_buffer = in MemoryUsageTracker()
851 for (const LogicalBuffer* logical_buffer : in MemoryUsageTracker()
Dheap_simulator.cc44 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule()
70 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
85 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation()
Dhlo_memory_scheduler_test.cc42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc169 TF_ASSIGN_OR_RETURN(const LogicalBuffer* call_result_buf, in AddBackendConstraintsToDnnConvCustomCall()
261 const LogicalBuffer* output_buffer; in AddBackendConstraints()
330 TF_ASSIGN_OR_RETURN(const LogicalBuffer* out_buf, in PropagateOperandConstraint()
345 TF_ASSIGN_OR_RETURN(const LogicalBuffer* out_buf, in PropagateOperandConstraint()
363 const LogicalBuffer& buf = buffer_constraint.buffer(); in PropagateBufferConstraint()
Dgpu_compiler.cc538 [](LogicalBuffer::Color) { return kXlaAllocatedBufferAlignBytes; }, in RunHloPassesAndBufferAssignement()
581 [](LogicalBuffer::Color) { return kXlaAllocatedBufferAlignBytes; }, in CompileModuleToLlvmIrImpl()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_noalias_test.cc69 [](LogicalBuffer::Color) { return /*alignment=*/1; }); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc32 /*index=*/-1, /*size=*/0, LogicalBuffer::Color(0));
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc467 int64 memory_alignment(LogicalBuffer::Color) { in memory_alignment()

12