Home
last modified time | relevance | path

Searched defs:compute_uv (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebraLib.cpp251 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 …r& U, const Tensor& S, const Tensor& V, const Tensor& infos, bool full_matrices, bool compute_uv) { in svd_cusolver_gesvdj()
442 const Tensor& infos, bool compute_uv in apply_svd_cusolver_gesvdjBatched()
484 …r& U, const Tensor& S, const Tensor& V, const Tensor& infos, bool full_matrices, bool compute_uv) { 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()
650 const bool compute_uv, in svd_cusolver()
H A DBatchLinearAlgebra.cpp2103 const bool compute_uv, in apply_svd_magma()
2172 const bool compute_uv, in svd_magma()
2212 const bool compute_uv, in svd_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBatchLinearAlgebraKernel.cpp1027 const bool compute_uv, in apply_svd()
1089 const bool compute_uv, in svd_kernel()
H A DBatchLinearAlgebra.cpp3224 std::tuple<Tensor&, Tensor&, Tensor&> svd_out(const Tensor& self, bool some, bool compute_uv, in svd_out()
3263 std::tuple<Tensor, Tensor, Tensor> svd(const Tensor& self, bool some, bool compute_uv) { in svd()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dlinalg_ops_test.cc253 auto set_attrs = [&op](bool compute_uv, bool full_matrices) { in TEST()
H A Dlinalg_ops.cc255 bool compute_uv; in SvdShapeFn() local
/aosp_15_r20/external/pytorch/torch/_numpy/
H A Dlinalg.py193 def svd(a: ArrayLike, full_matrices=True, compute_uv=True, hermitian=False): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dlinalg_ops.py488 def svd(tensor, full_matrices=False, compute_uv=True, name=None): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DLinearAlgebraStubs.cpp100 const bool compute_uv, in lazy_svd_kernel()