Searched refs:cuda_memory (Results 1 – 7 of 7) 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()
38 inline se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory, uint64 size) { in AsDeviceMemory() argument39 se::DeviceMemoryBase wrapped(const_cast<T*>(cuda_memory), size * sizeof(T)); in AsDeviceMemory()
37 se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* cuda_memory) { in AsDeviceMemory() argument38 se::DeviceMemoryBase wrapped(const_cast<Scalar*>(cuda_memory)); in AsDeviceMemory()
293 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory) { in AsDeviceMemory() argument294 se::DeviceMemoryBase wrapped(const_cast<T*>(cuda_memory)); in AsDeviceMemory()300 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory, uint64 size) { in AsDeviceMemory() argument301 se::DeviceMemoryBase wrapped(const_cast<T*>(cuda_memory), size * sizeof(T)); in AsDeviceMemory()
132 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory, uint64 size) { in AsDeviceMemory() argument133 se::DeviceMemoryBase wrapped(const_cast<T*>(cuda_memory), size * sizeof(T)); in AsDeviceMemory()
250 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory) {251 se::DeviceMemoryBase wrapped(const_cast<T*>(cuda_memory));
232 static se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* cuda_memory) { in AsDeviceMemory() argument233 se::DeviceMemoryBase wrapped(const_cast<Scalar*>(cuda_memory)); in AsDeviceMemory()