Searched refs:CuSparseMatDescriptor (Results 1 – 2 of 2) sorted by relevance
80 class TORCH_CUDA_CPP_API CuSparseMatDescriptor83 CuSparseMatDescriptor() { in CuSparseMatDescriptor() function89 CuSparseMatDescriptor(bool upper, bool unit) { in CuSparseMatDescriptor() function
172 auto desc = at::cuda::sparse::CuSparseMatDescriptor(upper, unitriangular); in block_sparse_triangular_solve_vec()308 auto desc = at::cuda::sparse::CuSparseMatDescriptor(upper, unitriangular); in block_sparse_triangular_solve_mat()433 auto desc = at::cuda::sparse::CuSparseMatDescriptor(); in block_sparse_mv()543 auto desc = at::cuda::sparse::CuSparseMatDescriptor(); in block_sparse_mm()1108 auto desc = at::cuda::sparse::CuSparseMatDescriptor(); in add_out_sparse_csr()