Home
last modified time | relevance | path

Searched defs:nrhs (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBatchLinearAlgebra.cpp807 template<> void lapackCholeskySolve<c10::complex<double>>(char uplo, int n, int nrhs, c10::complex<… in lapackCholeskySolve()
811 template<> void lapackCholeskySolve<c10::complex<float>>(char uplo, int n, int nrhs, c10::complex<f… in lapackCholeskySolve()
815 template<> void lapackCholeskySolve<double>(char uplo, int n, int nrhs, double *a, int lda, double … in lapackCholeskySolve()
819 template<> void lapackCholeskySolve<float>(char uplo, int n, int nrhs, float *a, int lda, float *b,… in lapackCholeskySolve()
1123 int nrhs, in lapackLdlSolveSymmetric()
1137 int nrhs, in lapackLdlSolveSymmetric()
1151 int nrhs, in lapackLdlSolveSymmetric()
1174 int nrhs, in lapackLdlSolveSymmetric()
1197 int nrhs, in lapackLdlSolveHermitian()
1211 int nrhs, in lapackLdlSolveHermitian()
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebra.cpp387 magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, double* dA, magma_int_t ldda, in magmaCholeskySolve()
396 magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, float* dA, magma_int_t ldda, in magmaCholeskySolve()
405 magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, c10::complex<double>* dA, magma_int_t ldda, in magmaCholeskySolve()
416 magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, c10::complex<float>* dA, magma_int_t ldda, in magmaCholeskySolve()
427 magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, double** dA_array, magma_int_t ldda, in magmaCholeskySolveBatched()
435 magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, float** dA_array, magma_int_t ldda, in magmaCholeskySolveBatched()
443 …magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, c10::complex<double>** dA_array, magma_int_t l… in magmaCholeskySolveBatched()
453 …magma_uplo_t uplo, magma_int_t n, magma_int_t nrhs, c10::complex<float>** dA_array, magma_int_t ld… in magmaCholeskySolveBatched()
869 magma_int_t n, magma_int_t nrhs, double* dA, magma_int_t ldda, magma_int_t* ipiv, in magmaLuSolve()
878 magma_int_t n, magma_int_t nrhs, float* dA, magma_int_t ldda, magma_int_t* ipiv, in magmaLuSolve()
[all …]
H A DCUDASolver.cpp87 …cusolverDnHandle_t handle, int n, int nrhs, double* dA, int lda, int* ipiv, double* ret, int ldb, … in getrs()
94 …cusolverDnHandle_t handle, int n, int nrhs, float* dA, int lda, int* ipiv, float* ret, int ldb, in… in getrs()
103 int nrhs, in getrs()
128 int nrhs, in getrs()
784 …cusolverDnHandle_t handle, cublasFillMode_t uplo, int n, int nrhs, const float *A, int lda, float … in potrs()
791 …cusolverDnHandle_t handle, cublasFillMode_t uplo, int n, int nrhs, const double *A, int lda, doubl… in potrs()
798 …cusolverDnHandle_t handle, cublasFillMode_t uplo, int n, int nrhs, const c10::complex<float> *A, i… in potrs()
810 …cusolverDnHandle_t handle, cublasFillMode_t uplo, int n, int nrhs, const c10::complex<double> *A, … in potrs()
822 …cusolverDnHandle_t handle, cublasFillMode_t uplo, int n, int nrhs, float *Aarray[], int lda, float… in potrsBatched()
829 …cusolverDnHandle_t handle, cublasFillMode_t uplo, int n, int nrhs, double *Aarray[], int lda, doub… in potrsBatched()
[all …]
H A DBatchLinearAlgebraLibBlas.cpp139 auto nrhs = cuda_int_cast(B.size(-1), "nrhs"); in apply_lu_solve_batched_cublas() local
246 auto nrhs = cuda_int_cast(B.size(-1), "nrhs"); in apply_gels_batched() local
H A DBatchLinearAlgebraLib.cpp827 const int64_t nrhs = self_working_copy.size(-1); in apply_cholesky_cusolver_potrs() local
883 const int64_t nrhs = self_working_copy.size(-1); in apply_cholesky_cusolver_potrsBatched() local
913 const int64_t nrhs = self_working_copy.size(-1); in _cholesky_solve_helper_cuda_cusolver() local
1507 int nrhs = cuda_int_cast(B.size(-1), "nrhs"); in lu_solve_looped_cusolver() local
/aosp_15_r20/external/eigen/Eigen/src/KLUSupport/
H A DKLUSupport.h34 inline int klu_solve(klu_symbolic *Symbolic, klu_numeric *Numeric, Index ldim, Index nrhs, double B… in klu_solve()
38 inline int klu_solve(klu_symbolic *Symbolic, klu_numeric *Numeric, Index ldim, Index nrhs, std::com… in klu_solve()
42 inline int klu_tsolve(klu_symbolic *Symbolic, klu_numeric *Numeric, Index ldim, Index nrhs, double … in klu_tsolve()
46 inline int klu_tsolve(klu_symbolic *Symbolic, klu_numeric *Numeric, Index ldim, Index nrhs, std::co… in klu_tsolve()
/aosp_15_r20/external/eigen/Eigen/src/SparseLU/
H A DSparseLU_SupernodalMatrix.h243 Index nrhs = Index(X.cols()); in solveInPlace() local
306 Index nrhs = Index(X.cols()); in solveTransposedInPlace() local
H A DSparseLU.h832 Index nrhs = X.cols(); in solveInPlace() local
875 Index nrhs = X.cols(); in solveTransposedInPlace() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dmatrix_solve_op_test.py54 def testSolve(self, n, nrhs, batch_dims, rhs_batch_dims, adjoint): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dmatrix_solve_op.cc133 const int64_t nrhs = rhs.dim_size(ndims - 1); in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dcuda_solvers.cc482 cublasOperation_t trans, int n, int nrhs, in GetrsImpl()
802 cublasHandle_t cublas_handle, cublasOperation_t trans, int n, int nrhs, in GetrsBatchedImpl()
/aosp_15_r20/external/cronet/third_party/boringssl/src/third_party/googletest/googlemock/test/
H A Dgmock-matchers-misc_test.cc552 size_t nrhs = GetParam().second; in TEST_P() local
/aosp_15_r20/external/googletest/googlemock/test/
H A Dgmock-matchers-misc_test.cc552 size_t nrhs = GetParam().second; in TEST_P() local
/aosp_15_r20/external/cronet/third_party/googletest/src/googlemock/test/
H A Dgmock-matchers-misc_test.cc552 size_t nrhs = GetParam().second; in TEST_P() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_linalg.py8367 def run_test(shape, batch, nrhs, hermitian): argument
/aosp_15_r20/external/stardoc/stardoc/
HDrenderer_binary.jarMETA-INF/ META-INF/MANIFEST.MF build-data.properties com/ com ...
HDstardoc_binary.jarMETA-INF/ META-INF/MANIFEST.MF build-data.properties com/ com ...
/aosp_15_r20/prebuilts/bazel/common/android_tools/android_tools/
Dall_android_tools_deploy.jarMETA-INF/ META-INF/MANIFEST.MF build-data.properties com/ com ...