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