Home
last modified time | relevance | path

Searched defs:cuda_memory (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfft_ops.cc355 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory) { in AsDeviceMemory()
362 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory, uint64 size) { in AsDeviceMemory()
H A Dgpu_utils.h73 inline se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory, uint64 size) { in AsDeviceMemory()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/
H A Dblas_gemm.cc29 se::DeviceMemory<T> AsDeviceMemory(const T* cuda_memory) { in AsDeviceMemory()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dtridiagonal_solve_op_gpu.cu.cc68 se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* cuda_memory) { in AsDeviceMemory()
/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A Dnccl_manager_test.cc439 static se::DeviceMemory<Scalar> AsDeviceMemory(const Scalar* cuda_memory) { in AsDeviceMemory()