Searched refs:AddAsyncCopy (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment.cc | 1585 AddAsyncCopy(*allocations.back(), MemorySpace::kAlternate, in AllocateCrossProgramPrefetchBuffer()
1597 AddAsyncCopy(*allocations.front(), MemorySpace::kAlternate, in AllocateCrossProgramPrefetchBuffer()
2111 void AlternateMemoryBestFitHeap::AddAsyncCopy( in AddAsyncCopy() function in xla::AlternateMemoryBestFitHeap
2344 AddAsyncCopy(*prev_allocation, MemorySpace::kDefault, in Evict()
2364 AddAsyncCopy(*prev_allocation, MemorySpace::kDefault, in Evict()
2516 AddAsyncCopy(prev_allocation_in_default_mem, MemorySpace::kAlternate, in Prefetch()
|
| D | memory_space_assignment.h | 1266 void AddAsyncCopy(const MemorySpaceAssignment::Allocation& prev_allocation,
|