Home
last modified time | relevance | path

Searched refs:nrhs (Results 1 – 25 of 27) sorted by relevance

12

/external/eigen/Eigen/src/misc/
Dlapacke.h272 lapack_int kl, lapack_int ku, lapack_int nrhs,
278 lapack_int kl, lapack_int ku, lapack_int nrhs,
284 lapack_int kl, lapack_int ku, lapack_int nrhs,
292 lapack_int kl, lapack_int ku, lapack_int nrhs,
302 lapack_int nrhs, const float* ab, lapack_int ldab,
312 lapack_int nrhs, const double* ab, lapack_int ldab,
322 lapack_int nrhs, const lapack_complex_float* ab,
333 lapack_int nrhs, const lapack_complex_double* ab,
344 lapack_int ku, lapack_int nrhs, float* ab,
348 lapack_int ku, lapack_int nrhs, double* ab,
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcusolver_dense_11_0.inc237 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
249 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
254 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
266 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
271 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
283 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
288 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
300 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
305 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
317 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
[all …]
Dcusolver_dense_10_2.inc245 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
257 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
262 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
274 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
279 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
291 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
296 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
307 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
312 cusolverDnHandle_t handle, cusolver_int_t n, cusolver_int_t nrhs,
323 return func_ptr(handle, n, nrhs, dA, ldda, dipiv, dB, lddb, dX, lddx,
[all …]
Dcusolver_dense_10_1.inc146 int nrhs, const float *A, int lda,
153 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
158 int nrhs, const double *A,
166 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
171 int nrhs, const cuComplex *A,
179 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
184 int nrhs,
193 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
246 int nrhs, /* only support rhs = 1*/
253 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, d_info, batchSize);
[all …]
Dcusolver_dense_10_0.inc139 int nrhs, const float *A, int lda,
146 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
151 int nrhs, const double *A,
159 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
164 int nrhs, const cuComplex *A,
172 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
177 int nrhs,
186 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
239 int nrhs, /* only support rhs = 1*/
246 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, d_info, batchSize);
[all …]
Dcusolver_dense_9_0.inc139 int nrhs, const float *A, int lda,
146 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
151 int nrhs, const double *A,
159 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
164 int nrhs, const cuComplex *A,
172 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
177 int nrhs,
186 return func_ptr(handle, uplo, n, nrhs, A, lda, B, ldb, devInfo);
320 int nrhs, const float *A, int lda,
328 return func_ptr(handle, trans, n, nrhs, A, lda, devIpiv, B, ldb, devInfo);
[all …]
Dcusparse_11_0.inc1529 cusparseOperation_t transB, int m, int nrhs, int nnz, const float *alpha,
1541 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
1548 cusparseOperation_t transB, int m, int nrhs, int nnz, const double *alpha,
1560 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
1567 cusparseOperation_t transB, int m, int nrhs, int nnz,
1579 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
1586 cusparseOperation_t transB, int m, int nrhs, int nnz,
1599 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
1606 cusparseOperation_t transB, int m, int nrhs, int nnz, const float *alpha,
1617 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
[all …]
Dcusparse_10_0.inc2351 cusparseOperation_t transA, cusparseOperation_t transB, int m, int nrhs,
2363 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2370 cusparseOperation_t transA, cusparseOperation_t transB, int m, int nrhs,
2382 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2389 cusparseOperation_t transA, cusparseOperation_t transB, int m, int nrhs,
2401 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2408 cusparseOperation_t transA, cusparseOperation_t transB, int m, int nrhs,
2421 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2428 cusparseOperation_t transA, cusparseOperation_t transB, int m, int nrhs,
2440 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
[all …]
Dcublas_9_0.inc2890 int nrhs, const float *Aarray[],
2899 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
2904 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
2912 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
2917 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
2925 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
2930 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
2938 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3101 int nrhs, float *Aarray[], /*Device pointer*/
3110 return func_ptr(handle, trans, m, n, nrhs, Aarray, lda, Carray, ldc, info,
[all …]
Dcusparse_10_1.inc2354 cusparseOperation_t transB, int m, int nrhs, int nnz, const float *alpha,
2366 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2373 cusparseOperation_t transB, int m, int nrhs, int nnz, const double *alpha,
2385 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2392 cusparseOperation_t transB, int m, int nrhs, int nnz,
2404 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2411 cusparseOperation_t transB, int m, int nrhs, int nnz,
2424 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2431 cusparseOperation_t transB, int m, int nrhs, int nnz, const float *alpha,
2442 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
[all …]
Dcusparse_10_2.inc2354 cusparseOperation_t transB, int m, int nrhs, int nnz, const float *alpha,
2366 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2373 cusparseOperation_t transB, int m, int nrhs, int nnz, const double *alpha,
2385 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2392 cusparseOperation_t transB, int m, int nrhs, int nnz,
2404 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2411 cusparseOperation_t transB, int m, int nrhs, int nnz,
2424 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
2431 cusparseOperation_t transB, int m, int nrhs, int nnz, const float *alpha,
2442 return func_ptr(handle, algo, transA, transB, m, nrhs, nnz, alpha, descrA,
[all …]
Dcublas_10_0.inc2957 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
2965 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
2970 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
2978 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
2983 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
2991 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
2996 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
3005 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3180 int nrhs, float *const Aarray[], /*Device pointer*/
3189 return func_ptr(handle, trans, m, n, nrhs, Aarray, lda, Carray, ldc, info,
[all …]
Dcublas_10_2.inc3082 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
3090 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3095 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
3103 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3108 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
3116 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3121 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
3130 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3305 int nrhs, float *const Aarray[], /*Device pointer*/
3314 return func_ptr(handle, trans, m, n, nrhs, Aarray, lda, Carray, ldc, info,
[all …]
Dcublas_10_1.inc3082 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
3090 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3095 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
3103 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3108 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
3116 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3121 cublasHandle_t handle, cublasOperation_t trans, int n, int nrhs,
3130 return func_ptr(handle, trans, n, nrhs, Aarray, lda, devIpiv, Barray, ldb,
3305 int nrhs, float *const Aarray[], /*Device pointer*/
3314 return func_ptr(handle, trans, m, n, nrhs, Aarray, lda, Carray, ldc, info,
[all …]
/external/eigen/lapack/
Dcholesky.cpp41 EIGEN_LAPACK_FUNC(potrs,(char* uplo, int *n, int *nrhs, RealScalar *pa, int *lda, RealScalar *pb, i…
46 else if(*nrhs<0) *info = -3;
58 MatrixType B(b,*n,*nrhs,*ldb);
Dlu.cpp46 EIGEN_LAPACK_FUNC(getrs,(char *trans, int *n, int *nrhs, RealScalar *pa, int *lda, int *ipiv, RealS…
51 else if(*nrhs<0) *info = -3;
63 MatrixType B(b,*n,*nrhs,*ldb);
/external/eigen/Eigen/src/PardisoSupport/
DPardisoSupport.h47 …IndexType *ia, IndexType *ja, IndexType *perm, IndexType nrhs, IndexType *iparm, IndexType msglvl,… in run()
50 …::pardiso(pt, &maxfct, &mnum, &type, &phase, &n, a, ia, ja, perm, &nrhs, iparm, &msglvl, b, x, &er… in run()
59 …IndexType *ia, IndexType *ja, IndexType *perm, IndexType nrhs, IndexType *iparm, IndexType msglvl,…
62 …::pardiso_64(pt, &maxfct, &mnum, &type, &phase, &n, a, ia, ja, perm, &nrhs, iparm, &msglvl, b, x, …
330 Index nrhs = Index(b.cols());
332 …eigen_assert(((MatrixBase<BDerived>::Flags & RowMajorBit) == 0 || nrhs == 1) && "Row-major right h…
333 …eigen_assert(((MatrixBase<XDerived>::Flags & RowMajorBit) == 0 || nrhs == 1) && "Row-major matrice…
334 eigen_assert(((nrhs == 1) || b.outerStride() == b.rows()));
359 … m_perm.data(), internal::convert_index<StorageIndex>(nrhs), m_iparm.data(), m_msglvl,
/external/eigen/Eigen/src/SparseLU/
DSparseLU_SupernodalMatrix.h240 Index nrhs = Index(X.cols()); in solveInPlace() local
242 Matrix<Scalar,Dynamic,Dest::ColsAtCompileTime, ColMajor> work(n, nrhs); // working vector in solveInPlace()
255 for (Index j = 0; j < nrhs; j++) in solveInPlace()
274 …ColsAtCompileTime, ColMajor>, 0, OuterStride<> > U (&(X(fsupc,0)), nsupc, nrhs, OuterStride<>(n) ); in solveInPlace()
282 for (Index j = 0; j < nrhs; j++) in solveInPlace()
DSparseLU.h731 Index nrhs = X.cols(); in solveInPlace() local
743 for (Index j = 0; j < nrhs; j++) in solveInPlace()
751 …ColsAtCompileTime, ColMajor>, 0, OuterStride<> > U (&(X(fsupc,0)), nsupc, nrhs, OuterStride<>(n) ); in solveInPlace()
755 for (Index j = 0; j < nrhs; ++j) in solveInPlace()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_solve_op.cc133 const int64 nrhs = rhs.dim_size(ndims - 1); in ComputeAsync() local
253 transposed_rhs_shape.AddDim(nrhs); in ComputeAsync()
261 if (nrhs > 1) { in ComputeAsync()
298 solver->GetrsBatched(adjoint_ ? CUBLAS_OP_C : CUBLAS_OP_T, n, nrhs, in ComputeAsync()
314 solver->Getrs(adjoint_ ? CUBLAS_OP_C : CUBLAS_OP_T, n, nrhs, in ComputeAsync()
324 if (nrhs > 1) { in ComputeAsync()
/external/tensorflow/tensorflow/python/kernel_tests/
Dmatrix_solve_op_test.py86 for nrhs in 1, 2, n:
87 rhs = self._generateMatrix(n, nrhs)
94 for nrhs in 1, n:
95 rhs = self._generateMatrix(n, nrhs)
/external/tensorflow/tensorflow/compiler/tests/
Dmatrix_solve_op_test.py56 def testSolve(self, n, nrhs, batch_dims, rhs_batch_dims, adjoint): argument
58 rhs = np.random.normal(-5.0, 5.0, rhs_batch_dims + [n, nrhs])
/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc485 cublasOperation_t trans, int n, int nrhs, in GetrsImpl() argument
490 TF_RETURN_IF_CUSOLVER_ERROR(solver(cusolver_dn_handle, trans, n, nrhs, in GetrsImpl()
499 cublasOperation_t trans, int n, int nrhs, const Scalar* A, int lda, \
502 cusolver_dn_handle_, trans, n, nrhs, A, lda, pivots, B, \
795 cublasHandle_t cublas_handle, cublasOperation_t trans, int n, int nrhs, in GetrsBatchedImpl() argument
816 cublas_handle, trans, n, nrhs, in GetrsBatchedImpl()
826 cublasOperation_t trans, int n, int nrhs, \
832 this, context_, cublas_handle_, trans, n, nrhs, \
Dcuda_solvers.h244 Status Getrs(cublasOperation_t trans, int n, int nrhs, const Scalar* A,
261 Status GetrsBatched(cublasOperation_t trans, int n, int nrhs,
/external/eigen/Eigen/src/QR/
DCompleteOrthogonalDecomposition.h472 const Index nrhs = rhs.cols();
474 Matrix<typename MatrixType::Scalar, Dynamic, 1> temp((std::max)(cols, nrhs));

12