Searched refs:CuSparseSpMatCsrDescriptor (Results 1 – 3 of 3) sorted by relevance
627 auto descA = at::cuda::sparse::CuSparseSpMatCsrDescriptor(mat1_32); in spmm()631 auto descA = at::cuda::sparse::CuSparseSpMatCsrDescriptor(mat1); in spmm()716 auto descA = at::cuda::sparse::CuSparseSpMatCsrDescriptor(A_32); in spgemm()717 auto descB = at::cuda::sparse::CuSparseSpMatCsrDescriptor(B_32); in spgemm()718 auto descC = at::cuda::sparse::CuSparseSpMatCsrDescriptor(C); in spgemm()996 auto descA = at::cuda::sparse::CuSparseSpMatCsrDescriptor(mat); in addmv_out_sparse_csr()1290 auto descA = at::cuda::sparse::CuSparseSpMatCsrDescriptor(A); in triangular_solve_out_sparse_csr()1451 auto descC = at::cuda::sparse::CuSparseSpMatCsrDescriptor(C, /*batch_offset=*/i); in sampled_addmm_out_sparse_csr()
141 CuSparseSpMatCsrDescriptor::CuSparseSpMatCsrDescriptor(const Tensor& input, int64_t batch_offset) { in CuSparseSpMatCsrDescriptor() function in at::cuda::sparse::CuSparseSpMatCsrDescriptor
199 class TORCH_CUDA_CPP_API CuSparseSpMatCsrDescriptor202 explicit CuSparseSpMatCsrDescriptor(const Tensor& input, int64_t batch_offset = -1);