Home
last modified time | relevance | path

Searched refs:MemorySpace (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc939 existing_required_assignment->memory_space == MemorySpace::kDefault) { in IsUseAllowedInAlternateMemory()
1067 if (allocation.memory_space() == MemorySpace::kAlternate) { in AppendAllocationInfoDebugString()
1264 MemorySpace::kDefault); in AddRequiredAssignmentsForColocatedIntervals()
1268 position.index, MemorySpace::kDefault); in AddRequiredAssignmentsForColocatedIntervals()
1378 MemorySpace::kDefault, use_time); in AllocateAllocationValues()
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()
1547 buffer->defining_position(), MemorySpace::kDefault, kDummyChunk, in AllocateCrossProgramPrefetchBuffer()
1585 AddAsyncCopy(*allocations.back(), MemorySpace::kAlternate, in AllocateCrossProgramPrefetchBuffer()
[all …]
Dmemory_space_assignment.h402 enum class MemorySpace { kDefault, kAlternate }; enum
524 Allocation(HloPosition defining_position, MemorySpace memory_space, in Allocation()
554 MemorySpace memory_space() const { return memory_space_; } in memory_space()
577 MemorySpace memory_space_;
589 CopyAllocation(const Allocation& prev_allocation, MemorySpace memory_space,
898 MemorySpaceAssignment::MemorySpace destination;
944 using MemorySpace = MemorySpaceAssignment::MemorySpace;
1055 MemorySpaceAssignment::MemorySpace memory_space;
1222 MemorySpace memory_space, int64 time,
1225 ShapeIndex index, MemorySpace memory_space,
[all …]
Dmemory_space_assignment_test.cc4800 auto alternate_mem_space = MemorySpaceAssignment::MemorySpace::kAlternate; in TEST_F()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.h49 using MemorySpace = gpu::MemorySpace; variable
Dcuda_driver.cc170 std::string MemorySpaceString(MemorySpace memory_space) { in MemorySpaceString()
172 case MemorySpace::kHost: in MemorySpaceString()
174 case MemorySpace::kDevice: in MemorySpaceString()
1299 /* static */ port::StatusOr<MemorySpace> GpuDriver::GetPointerMemorySpace( in GetPointerMemorySpace()
1307 return MemorySpace::kDevice; in GetPointerMemorySpace()
1309 return MemorySpace::kHost; in GetPointerMemorySpace()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h34 enum class MemorySpace { kHost, kDevice }; enum
37 std::string MemorySpaceString(MemorySpace memory_space);
451 static port::StatusOr<MemorySpace> GetPointerMemorySpace(
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc133 string MemorySpaceString(MemorySpace memory_space) { in MemorySpaceString()
135 case MemorySpace::kHost: in MemorySpaceString()
137 case MemorySpace::kDevice: in MemorySpaceString()
1023 /* static */ port::StatusOr<MemorySpace> GpuDriver::GetPointerMemorySpace( in GetPointerMemorySpace()
1030 return MemorySpace::kDevice; in GetPointerMemorySpace()
1032 return MemorySpace::kHost; in GetPointerMemorySpace()