Searched refs:gpu_memory (Results 1 – 2 of 2) sorted by relevance
53 se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* gpu_memory) { in AsDeviceMemory() argument54 se::DeviceMemoryBase wrapped(const_cast<Scalar*>(gpu_memory)); in AsDeviceMemory()
282 se::DeviceMemory<T> AsDeviceMemory(const T* gpu_memory) {283 se::DeviceMemoryBase wrapped(const_cast<T*>(gpu_memory));