Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc827 ChunkCandidate chunk_candidate = FindChunkCandidate(buffer_interval); in Finish() local
830 CommitChunk(buffer_interval, chunk_candidate); in Finish()
898 ChunkCandidate chunk_candidate{Chunk{-1, INT64_MAX}, result_.heap_size}; in FindChunkCandidate() local
899 Chunk& min_fit_chunk = chunk_candidate.chunk; in FindChunkCandidate()
915 chunk_candidate.heap_size = preferred_chunk_end; in FindChunkCandidate()
937 chunk_candidate.heap_size = preferred_chunk_end; in FindChunkCandidate()
943 chunk_candidate.heap_size = offset + buffer_interval.size; in FindChunkCandidate()
948 return chunk_candidate; in FindChunkCandidate()
956 chunk_candidate) { in CommitChunk()
959 result_.heap_size = chunk_candidate.heap_size; in CommitChunk()
[all …]
Dmemory_space_assignment.cc1532 ChunkCandidate chunk_candidate = FindChunkCandidate(*prefetch_candidate); in AllocateCrossProgramPrefetchBuffer() local
1533 if (chunk_candidate.chunk.offset != 0 || in AllocateCrossProgramPrefetchBuffer()
1534 chunk_candidate.heap_size > available_heap_size()) { in AllocateCrossProgramPrefetchBuffer()
1539 AddToPendingChunks(*prefetch_candidate, chunk_candidate); in AllocateCrossProgramPrefetchBuffer()
1586 chunk_candidate.chunk, prefetch_candidate->start, in AllocateCrossProgramPrefetchBuffer()
1598 chunk_candidate.chunk, end_of_program_prefetch_start_time, in AllocateCrossProgramPrefetchBuffer()
1931 const ChunkCandidate& chunk_candidate) { in AddToPendingChunks() argument
1933 << buffer_interval.end << " : [" << chunk_candidate.chunk.offset in AddToPendingChunks()
1934 << ", " << chunk_candidate.chunk.size << "]"; in AddToPendingChunks()
1935 pending_chunks_.emplace_back(buffer_interval, chunk_candidate); in AddToPendingChunks()
[all …]
Dheap_simulator.h428 ChunkCandidate chunk_candidate);
Dmemory_space_assignment.h1278 const ChunkCandidate& chunk_candidate);