Searched refs:chunk_candidate (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment.cc | 2108 Chunk chunk_candidate = FindChunkCandidate(*prefetch_candidate); in AllocateCrossProgramPrefetchBuffer() local
2109 if (chunk_candidate.chunk_end() > available_heap_size()) { in AllocateCrossProgramPrefetchBuffer()
2187 AddAsyncCopy(*allocations.back(), MemorySpace::kAlternate, chunk_candidate, in AllocateCrossProgramPrefetchBuffer()
2206 AddAsyncCopy(*allocations.front(), MemorySpace::kAlternate, chunk_candidate, in AllocateCrossProgramPrefetchBuffer()
2224 AddToPendingChunks(buffer_interval, chunk_candidate); in AllocateCrossProgramPrefetchBuffer()
2270 Chunk chunk_candidate = in AllocateReservedScopedAllocations() local
2272 CHECK_EQ(chunk_candidate.offset, 0); in AllocateReservedScopedAllocations()
2273 AddToPendingChunks(interval, chunk_candidate); in AllocateReservedScopedAllocations()
2283 chunk_candidate, i, i, /*is_scoped_allocation=*/true)); in AllocateReservedScopedAllocations()
2670 const BufferInterval& buffer_interval, const Chunk& chunk_candidate) { in AddToPendingChunks() argument
[all …]
|
| D | heap_simulator.cc | 1005 Chunk chunk_candidate = FindChunkCandidate(buffer_interval); in Finish() local
1006 if (chunk_candidate.chunk_end() <= size_limit_per_heap_ || in Finish()
1012 CommitChunk(buffer_interval, chunk_candidate); in Finish()
|