| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebraLib.cpp | 251 const Tensor& infos, bool full_matrices, bool compute_uv, in apply_svd_cusolver_gesvd()
323 const Tensor& infos, bool full_matrices, bool compute_uv, in svd_cusolver_gesvd()
360 const Tensor& infos, bool full_matrices, bool compute_uv) { in apply_svd_cusolver_gesvdj()
433 … Tensor& U, const Tensor& S, const Tensor& V, const Tensor& infos, bool full_matrices, bool comput… in svd_cusolver_gesvdj()
484 … Tensor& U, const Tensor& S, const Tensor& V, const Tensor& infos, bool full_matrices, bool comput… in svd_cusolver_gesvdjBatched()
524 const Tensor& infos, bool full_matrices, bool compute_uv) { in apply_svd_cusolver_gesvdaStridedBatched()
582 const Tensor& infos, bool full_matrices, bool compute_uv) { in svd_cusolver_gesvdaStridedBatched()
649 const bool full_matrices, in svd_cusolver()
|
| H A D | BatchLinearAlgebra.cpp | 2102 const bool full_matrices, in apply_svd_magma()
2171 const bool full_matrices, in svd_magma()
2211 const bool full_matrices, in svd_kernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | linalg_ops_test.cc | 202 auto set_attrs = [&op](bool full_matrices) { in TEST()
253 auto set_attrs = [&op](bool compute_uv, bool full_matrices) { in TEST()
|
| H A D | linalg_ops.cc | 223 bool full_matrices; in QrShapeFn() local
260 bool full_matrices; in SvdShapeFn() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | linalg.h | 384 bool full_matrices, in svd()
394 bool full_matrices, in svd_out()
978 bool full_matrices, in svd()
988 bool full_matrices, in svd_out()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebraKernel.cpp | 1026 const bool full_matrices, in apply_svd()
1088 const bool full_matrices, in svd_kernel()
|
| H A D | BatchLinearAlgebra.cpp | 3185 bool full_matrices, in linalg_svd_out()
3204 std::tuple<Tensor, Tensor, Tensor> linalg_svd(const Tensor& A, bool full_matrices, in linalg_svd()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | qr.cc | 110 void QrExplicit(XlaOp a, bool full_matrices, XlaOp& q, XlaOp& r) { in QrExplicit()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | qr_op_test.py | 84 def _test(self, x_np, full_matrices, full_rank=True): argument
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | linalg.py | 193 def svd(a: ArrayLike, full_matrices=True, compute_uv=True, hermitian=False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | ops.cc | 287 [](XlaOp a, bool full_matrices) -> StatusOr<std::pair<XlaOp, XlaOp>> { in BuildOpsSubmodule()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | linalg_ops.py | 488 def svd(tensor, full_matrices=False, compute_uv=True, name=None): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LinearAlgebraStubs.cpp | 99 const bool full_matrices, in lazy_svd_kernel()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 3389 const bool full_matrices) { in linalg_svd_jvp()
|