Searched refs:CuSparseDnMatDescriptor (Results 1 – 3 of 3) sorted by relevance
131 class TORCH_CUDA_CPP_API CuSparseDnMatDescriptor134 explicit CuSparseDnMatDescriptor(const Tensor& input, int64_t batch_offset = -1);159 class TORCH_CUDA_CPP_API CuSparseDnMatDescriptor164 explicit CuSparseDnMatDescriptor(
115 CuSparseDnMatDescriptor::CuSparseDnMatDescriptor(const Tensor& input, int64_t batch_offset) { in CuSparseDnMatDescriptor() function in at::cuda::sparse::CuSparseDnMatDescriptor
637 auto descC = at::cuda::sparse::CuSparseDnMatDescriptor(*result_); in spmm()1363 auto descB = at::cuda::sparse::CuSparseDnMatDescriptor(*B_); in triangular_solve_out_sparse_csr()1364 auto descX = at::cuda::sparse::CuSparseDnMatDescriptor(*X_); in triangular_solve_out_sparse_csr()