Home
last modified time | relevance | path

Searched refs:buffer_interval (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc615 BufferInterval& buffer_interval = FindOrDie(buffer_intervals_, buffer); in Free() local
616 DCHECK_EQ(buffer_interval.buffer, buffer); in Free()
617 DCHECK_EQ(buffer_interval.size, size); in Free()
618 DCHECK_EQ(buffer_interval.end, -1); in Free()
619 if (buffer_interval.end != -1) { in Free()
622 buffer_interval.end = current_time_; in Free()
822 for (auto& buffer_interval : sorted_buffer_intervals) { in Finish() local
823 if (!buffer_interval.need_allocation) { in Finish()
827 ChunkCandidate chunk_candidate = FindChunkCandidate(buffer_interval); in Finish()
830 CommitChunk(buffer_interval, chunk_candidate); in Finish()
[all …]
Dheap_simulator.h425 ChunkCandidate FindChunkCandidate(const BufferInterval& buffer_interval,
427 void CommitChunk(const BufferInterval& buffer_interval,
Dmemory_space_assignment.cc742 const AlternateMemoryBestFitHeap::BufferInterval& buffer_interval, in CreateAllocationValues() argument
744 const HloValue* value = buffer_interval.buffer; in CreateAllocationValues()
775 allocation_values.emplace_back(value, position, buffer_interval.size); in CreateAllocationValues()
1930 const BufferInterval& buffer_interval, in AddToPendingChunks() argument
1932 VLOG(3) << "Committing chunk: " << buffer_interval.start << "-" in AddToPendingChunks()
1933 << buffer_interval.end << " : [" << chunk_candidate.chunk.offset in AddToPendingChunks()
1935 pending_chunks_.emplace_back(buffer_interval, chunk_candidate); in AddToPendingChunks()
1936 CommitChunk(buffer_interval, chunk_candidate); in AddToPendingChunks()
Dmemory_space_assignment.h981 const BufferInterval& buffer_interval,
1277 void AddToPendingChunks(const BufferInterval& buffer_interval,