Home
last modified time | relevance | path

Searched refs:kAlternate (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc1067 if (allocation.memory_space() == MemorySpace::kAlternate) { in AppendAllocationInfoDebugString()
1448 aliased_allocation->memory_space() == MemorySpace::kAlternate) { in AllocateAllocationValues()
1481 {start_time, end_time, MemorySpaceAssignment::MemorySpace::kAlternate}); in ViolatesOrdering()
1502 CHECK(allocation.memory_space() == MemorySpace::kAlternate); in CreateOrAddToAliasedOffset()
1585 AddAsyncCopy(*allocations.back(), MemorySpace::kAlternate, in AllocateCrossProgramPrefetchBuffer()
1597 AddAsyncCopy(*allocations.front(), MemorySpace::kAlternate, in AllocateCrossProgramPrefetchBuffer()
1614 if (allocation->memory_space() == MemorySpace::kAlternate) { in AllocateCrossProgramPrefetchBuffer()
1678 if (aliased_allocation->memory_space() == MemorySpace::kAlternate) { in AddAliasedRequiredAssignment()
1737 memory_space = MemorySpace::kAlternate; in AddInputAndOutputRequiredAssignments()
1761 memory_space = MemorySpace::kAlternate; in AddInputAndOutputRequiredAssignments()
[all …]
Dmemory_space_assignment.h402 enum class MemorySpace { kDefault, kAlternate }; enumerator
Dmemory_space_assignment_test.cc4800 auto alternate_mem_space = MemorySpaceAssignment::MemorySpace::kAlternate; in TEST_F()