/external/tensorflow/tensorflow/compiler/xla/service/ |
D | buffer_assignment.h | 71 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 …]
|
D | buffer_assignment.cc | 65 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 …]
|
D | hlo_memory_scheduler.h | 40 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_;
|
D | buffer_liveness.h | 48 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_;
|
D | tuple_points_to_analysis.h | 72 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 …]
|
D | logical_buffer_analysis.h | 34 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;
|
D | hlo_memory_scheduler.cc | 80 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 …]
|
D | logical_buffer.cc | 26 LogicalBuffer::LogicalBuffer(HloInstruction* instruction, in LogicalBuffer() function in xla::LogicalBuffer 32 LogicalBuffer::~LogicalBuffer() {} in ~LogicalBuffer() 34 string LogicalBuffer::ToString() const { in ToString()
|
D | buffer_liveness.cc | 56 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()
|
D | logical_buffer.h | 35 class LogicalBuffer : public BufferValue { 37 LogicalBuffer(HloInstruction* instruction, const ShapeIndex& index, Id id); 38 ~LogicalBuffer() override; 62 TF_DISALLOW_COPY_AND_ASSIGN(LogicalBuffer);
|
D | layout_assignment.h | 74 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_;
|
D | tuple_points_to_analysis.cc | 59 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 …]
|
D | logical_buffer_analysis.cc | 78 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()
|
D | buffer_value_containers.h | 35 for (const LogicalBuffer* buffer : logical_buffer_container) { in ToBufferValueCompactPointerSet() 47 for (const LogicalBuffer* buffer : logical_buffer_container) { in ToBufferValueFlatSet()
|
D | layout_assignment.cc | 66 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 …]
|
D | buffer_assignment_test.cc | 94 [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 …]
|
D | hlo.proto | 347 // 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
|
D | heap_simulator.h | 95 const LogicalBuffer::SizeFunction& size_function); 102 const LogicalBuffer::SizeFunction& size_function,
|
D | hlo_rematerialization.cc | 290 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()
|
D | tuple_points_to_analysis_test.cc | 64 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/ |
D | alias_analysis.cc | 31 /*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()
|
D | buffer_assignment_util.cc | 26 const LogicalBuffer* buffer = buffer_offset_pair.first; in InstrForConstantBufferAllocation()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_layout_assignment.cc | 147 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/ |
D | cpu_noalias_test.cc | 67 [](LogicalBuffer::Color) { return /*alignment=*/1; }); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_executable.cc | 228 const LogicalBuffer* buffer_source = sources[0]; in CreateResultShapedBuffer()
|