/external/eigen/Eigen/src/misc/ |
D | lapacke.h | 272 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/ |
D | cusolver_dense_11_0.inc | 237 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 …]
|
D | cusolver_dense_10_2.inc | 245 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 …]
|
D | cusolver_dense_10_1.inc | 146 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 …]
|
D | cusolver_dense_10_0.inc | 139 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 …]
|
D | cusolver_dense_9_0.inc | 139 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 …]
|
D | cusparse_11_0.inc | 1529 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 …]
|
D | cusparse_10_0.inc | 2351 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 …]
|
D | cublas_9_0.inc | 2890 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 …]
|
D | cusparse_10_2.inc | 2354 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 …]
|
D | cusparse_10_1.inc | 2354 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 …]
|
D | cublas_11_0.inc | 3082 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 …]
|
D | cublas_10_1.inc | 3082 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 …]
|
D | cublas_10_0.inc | 2957 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 …]
|
/external/eigen/lapack/ |
D | cholesky.cpp | 41 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);
|
D | lu.cpp | 46 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/ |
D | PardisoSupport.h | 47 …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/ |
D | SparseLU_SupernodalMatrix.h | 240 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()
|
D | SparseLU.h | 731 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/ |
D | matrix_solve_op.cc | 133 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/ |
D | matrix_solve_op_test.py | 86 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/ |
D | matrix_solve_op_test.py | 56 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/ |
D | cuda_solvers.cc | 485 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, \
|
D | cuda_solvers.h | 244 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/ |
D | CompleteOrthogonalDecomposition.h | 472 const Index nrhs = rhs.cols(); 474 Matrix<typename MatrixType::Scalar, Dynamic, 1> temp((std::max)(cols, nrhs));
|