Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.h49 using MemorySpace = gpu::MemorySpace; variable
Dcuda_driver.cc156 string MemorySpaceString(MemorySpace memory_space) { in MemorySpaceString()
158 case MemorySpace::kHost: in MemorySpaceString()
160 case MemorySpace::kDevice: in MemorySpaceString()
1294 /* static */ port::StatusOr<MemorySpace> GpuDriver::GetPointerMemorySpace( in GetPointerMemorySpace()
1302 return MemorySpace::kDevice; in GetPointerMemorySpace()
1304 return MemorySpace::kHost; in GetPointerMemorySpace()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h37 enum class MemorySpace { kHost, kDevice }; enum
40 string MemorySpaceString(MemorySpace memory_space);
385 static port::StatusOr<MemorySpace> GetPointerMemorySpace(
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc132 string MemorySpaceString(MemorySpace memory_space) { in MemorySpaceString()
134 case MemorySpace::kHost: in MemorySpaceString()
136 case MemorySpace::kDevice: in MemorySpaceString()
1086 /* static */ port::StatusOr<MemorySpace> GpuDriver::GetPointerMemorySpace( in GetPointerMemorySpace()
1093 return MemorySpace::kDevice; in GetPointerMemorySpace()
1095 return MemorySpace::kHost; in GetPointerMemorySpace()