Home
last modified time | relevance | path

Searched refs:sparse_sparse_matmul_cuda_kernel (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseMatMul.cu666 void sparse_sparse_matmul_cuda_kernel( in sparse_sparse_matmul_cuda_kernel() function
815 sparse_sparse_matmul_cuda_kernel<scalar_t>(output, mat1_.coalesce(), mat2_.coalesce()); in sparse_sparse_matmul_cuda()
819 sparse_sparse_matmul_cuda_kernel<scalar_t>(output, mat1_.coalesce(), mat2_.coalesce()); in sparse_sparse_matmul_cuda()