Searched refs:cuda_memory (Results 1 – 5 of 5) sorted by relevance
29 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory) { in AsDeviceMemory() argument30 se::DeviceMemoryBase wrapped(const_cast<T*>(cuda_memory)); in AsDeviceMemory()
63 inline se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory, uint64 size) { in AsDeviceMemory() argument64 se::DeviceMemoryBase wrapped(const_cast<T*>(cuda_memory), size * sizeof(T)); in AsDeviceMemory()
344 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory) { in AsDeviceMemory() argument345 se::DeviceMemoryBase wrapped(const_cast<T*>(cuda_memory)); in AsDeviceMemory()351 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory, uint64 size) { in AsDeviceMemory() argument352 se::DeviceMemoryBase wrapped(const_cast<T*>(cuda_memory), size * sizeof(T)); in AsDeviceMemory()
71 se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* cuda_memory) { in AsDeviceMemory() argument72 se::DeviceMemoryBase wrapped(const_cast<Scalar*>(cuda_memory)); in AsDeviceMemory()
439 static se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* cuda_memory) { in AsDeviceMemory() argument440 se::DeviceMemoryBase wrapped(const_cast<Scalar*>(cuda_memory)); in AsDeviceMemory()