Searched refs:buffer_interval (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator.cc | 602 BufferInterval& buffer_interval = FindOrDie(buffer_intervals_, buffer); in Free() local
603 DCHECK_EQ(buffer_interval.buffer, buffer); in Free()
604 DCHECK_EQ(buffer_interval.size, size); in Free()
605 DCHECK_EQ(buffer_interval.end, -1); in Free()
606 if (buffer_interval.end != -1) { in Free()
609 buffer_interval.end = current_time_; in Free()
810 for (auto& buffer_interval : sorted_buffer_intervals) { in Finish() local
811 if (!buffer_interval.need_allocation) { in Finish()
817 CommitChunk(buffer_interval, FindChunkCandidate(buffer_interval)); in Finish()
843 const GlobalDecreasingSizeBestFitHeap::BufferInterval& buffer_interval, in FindChunkCandidate() argument
[all …]
|
| D | heap_simulator.h | 423 Chunk FindChunkCandidate(const BufferInterval& buffer_interval,
425 void CommitChunk(const BufferInterval& buffer_interval, Chunk chunk);
|
| D | memory_space_assignment.cc | 1009 const AlternateMemoryBestFitHeap::BufferInterval& buffer_interval, in CreateAllocationValues() argument
1011 const HloValue* value = buffer_interval.buffer; in CreateAllocationValues()
1042 allocation_values.emplace_back(value, position, buffer_interval.size); in CreateAllocationValues()
2219 BufferInterval buffer_interval; in AllocateCrossProgramPrefetchBuffer() local
2220 buffer_interval.start = allocation->start_time(); in AllocateCrossProgramPrefetchBuffer()
2221 buffer_interval.end = allocation->end_time(); in AllocateCrossProgramPrefetchBuffer()
2222 buffer_interval.size = allocation->chunk().size; in AllocateCrossProgramPrefetchBuffer()
2223 buffer_interval.buffer = prefetch_candidate->buffer; in AllocateCrossProgramPrefetchBuffer()
2224 AddToPendingChunks(buffer_interval, chunk_candidate); in AllocateCrossProgramPrefetchBuffer()
2670 const BufferInterval& buffer_interval, const Chunk& chunk_candidate) { in AddToPendingChunks() argument
[all …]
|
| D | memory_space_assignment.h | 1243 const BufferInterval& buffer_interval,
1539 void AddToPendingChunks(const BufferInterval& buffer_interval,
|