Searched defs:AsDeviceMemory (Results 1 – 9 of 9) sorted by relevance
33 static se::DeviceMemory<T> AsDeviceMemory(const Tensor& t) { in AsDeviceMemory() function
73 static se::DeviceMemory<T> AsDeviceMemory(const Tensor& t) { in AsDeviceMemory() function
45 se::DeviceMemory<T> AsDeviceMemory(const T* gpu_memory) { in AsDeviceMemory() function73 inline se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory, uint64 size) { in AsDeviceMemory() function
355 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory) { in AsDeviceMemory() function362 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory, uint64 size) { in AsDeviceMemory() function
296 const DeviceMemory<T> AsDeviceMemory(const Tensor* tensor) { in AsDeviceMemory() function303 DeviceMemory<T> AsDeviceMemory(Tensor* tensor) { in AsDeviceMemory() function
29 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory) { in AsDeviceMemory() function
48 se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* gpu_memory) { in AsDeviceMemory() function
68 se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* cuda_memory) { in AsDeviceMemory() function
439 static se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* cuda_memory) { in AsDeviceMemory() function in tensorflow::NcclManagerTest