Searched refs:prev_allocation (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment.cc | 2112 const MemorySpaceAssignment::Allocation& prev_allocation, in AddAsyncCopy() argument
2127 prev_allocation, memory_space, chunk, start_time, end_time, in AddAsyncCopy()
2180 MemorySpaceAssignment::Allocation* prev_allocation = nullptr; in AllocateInAlternateMemoryNoCopy() local
2190 prev_allocation = in AllocateInAlternateMemoryNoCopy()
2193 (prev_allocation->memory_space() == MemorySpace::kAlternate); in AllocateInAlternateMemoryNoCopy()
2216 if (prev_allocation != nullptr) { in AllocateInAlternateMemoryNoCopy()
2217 preferred_offset = GetAliasedOffset(*prev_allocation); in AllocateInAlternateMemoryNoCopy()
2220 alternate_mem_interval.start = prev_allocation->end_time() + 1; in AllocateInAlternateMemoryNoCopy()
2272 if (prev_allocation != nullptr && in AllocateInAlternateMemoryNoCopy()
2273 (prev_allocation->is_copy_allocation() || in AllocateInAlternateMemoryNoCopy()
[all …]
|
| D | memory_space_assignment.h | 589 CopyAllocation(const Allocation& prev_allocation, MemorySpace memory_space,
595 prev_allocation_(prev_allocation),
1266 void AddAsyncCopy(const MemorySpaceAssignment::Allocation& prev_allocation,
|