def _solveM(self, rhs):
A = self.A
return TriSolve.forward(A.data, A.indptr, A.indices, rhs, self.A.nnz, *rhs.shape)
def _solveM(self, rhs):
A = self.A
return TriSolve.forward(A.data, A.indptr, A.indices, rhs, self.A.nnz,
*rhs.shape)
def _solve1(self, rhs):
A = self.A
x = TriSolve.forward(A.data, A.indptr, A.indices, rhs, self.A.nnz, rhs.size, 1)
return x.flatten()
def _solve1(self, rhs):
A = self.A
x = TriSolve.forward(A.data, A.indptr, A.indices, rhs, self.A.nnz,
rhs.size, 1)
return x.flatten()
