Home
last modified time | relevance | path

Searched refs:live_buffers (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc121 live_buffers; in RunComputation() local
124 auto add_user_to_buffer = [this, &live_buffers, &used_buffers]( in RunComputation()
130 live_buffers[buffer].insert(user); in RunComputation()
197 live_buffers[buffer].insert(nullptr); in RunComputation()
202 if (!live_buffers.contains(buffer)) { in RunComputation()
219 auto it = live_buffers.find(operand_buffer); in RunComputation()
223 live_buffers.erase(it); in RunComputation()
329 to_free.reserve(live_buffers.size()); in RunComputation()
330 for (const auto& buffer_pending : live_buffers) { in RunComputation()
Dbuffer_assignment.cc1288 absl::flat_hash_set<const LogicalBuffer*> live_buffers; in ComputePeakMemoryLogicalBuffers() local
1293 InsertOrDie(&live_buffers, buffer); in ComputePeakMemoryLogicalBuffers()
1297 CHECK(ContainsKey(live_buffers, buffer)); in ComputePeakMemoryLogicalBuffers()
1298 live_buffers.erase(buffer); in ComputePeakMemoryLogicalBuffers()
1309 live_buffers_vector.insert(live_buffers_vector.end(), live_buffers.begin(), in ComputePeakMemoryLogicalBuffers()
1310 live_buffers.end()); in ComputePeakMemoryLogicalBuffers()