Searched refs:CopyFromGpuDeviceArrayToLocal (Results 1 – 1 of 1) sorted by relevance
383 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void CopyFromGpuDeviceArrayToLocal( in CopyFromGpuDeviceArrayToLocal() function404 CopyFromGpuDeviceArrayToLocal(std::move(batch_ptr_s), local_batch_ptr, in CSRSparseMatrixSoftmaxKernel3D()577 CopyFromGpuDeviceArrayToLocal(std::move(softmax_and_grad_batch_ptr_s), in CSRSparseMatrixSoftmaxGradKernel3D()