| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebra.cpp | 1322 double *s, double rcond, int *rank, in lapackGelsd()
1336 float *s, float rcond, int *rank, in lapackGelsd()
1350 double *s, double rcond, int *rank, in lapackGelsd()
1362 float *s, float rcond, int *rank, in lapackGelsd()
1374 int *jpvt, double rcond, int *rank, in lapackGelsy()
1387 int *jpvt, float rcond, int *rank, in lapackGelsy()
1400 int *jpvt, double rcond, int *rank, in lapackGelsy()
1411 int *jpvt, float rcond, int *rank, in lapackGelsy()
1422 double *s, double rcond, int *rank, in lapackGelss()
1437 float *s, float rcond, int *rank, in lapackGelss()
[all …]
|
| H A D | BatchLinearAlgebraKernel.cpp | 471 …& A, Tensor& B, Tensor& rank, Tensor& singular_values, Tensor& infos, double rcond, LapackLstsqDri… in apply_lstsq()
610 …& a, Tensor& b, Tensor& rank, Tensor& singular_values, Tensor& infos, double rcond, std::string dr… in lstsq_kernel()
|
| H A D | LinearAlgebra.cpp | 557 Tensor linalg_pinv(const Tensor& input, const Tensor& rcond, bool hermitian) { in linalg_pinv()
569 Tensor linalg_pinv(const Tensor& input, double rcond, bool hermitian) { in linalg_pinv()
603 Tensor& linalg_pinv_out(const Tensor& input, const Tensor& rcond, bool hermitian, Tensor& result) { in linalg_pinv_out()
613 Tensor& linalg_pinv_out(const Tensor& input, double rcond, bool hermitian, Tensor& result) { in linalg_pinv_out()
618 Tensor pinverse(const Tensor& self, double rcond) { in pinverse()
|
| /aosp_15_r20/external/eigen/test/ |
| H A D | lu.cpp | 144 RealScalar rcond = (RealScalar(1) / matrix_l1_norm(m1)) / matrix_l1_norm(m1_inverse); in lu_invertible() local
177 RealScalar rcond = (RealScalar(1) / matrix_l1_norm(m1)) / matrix_l1_norm(m1_inverse); in lu_partial_piv() local
|
| H A D | cholesky.cpp | 94 RealScalar rcond = (RealScalar(1) / matrix_l1_norm<MatrixType, Lower>(symmLo)) / in cholesky() local
166 RealScalar rcond = (RealScalar(1) / matrix_l1_norm<MatrixType, Lower>(symmLo)) / in cholesky() local
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | linalg.py | 74 def lstsq(a: ArrayLike, b: ArrayLike, rcond=None): argument
92 def pinv(a: ArrayLike, rcond=1e-15, hermitian=False): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LinearAlgebraStubs.cpp | 115 …& a, Tensor& b, Tensor& rank, Tensor& singular_values, Tensor& infos, double rcond, std::string dr… in lazy_lstsq_kernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A D | linalg_ops_test.py | 328 def expected_pinv(self, a, rcond): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | linalg.h | 304 inline Tensor pinv(const Tensor& input, double rcond, bool hermitian) { in pinv()
311 double rcond, in pinv_out()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/ |
| H A D | linalg_impl.py | 806 def pinv(a, rcond=None, validate_args=False, name=None): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesLinearAlgebra.cpp | 430 std::optional<double> rcond, std::optional<c10::string_view> driver) { in linalg_lstsq_batch_rule()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_linalg.py | 287 def check_correctness_numpy(a, b, res, driver, rcond): argument
|