Searched defs:compute_eigenvectors (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebraKernel.cpp | 146 …alg_eig(Tensor& values, Tensor& vectors, Tensor& input, Tensor& infos, bool compute_eigenvectors) { in apply_linalg_eig()
197 …eigenvalues, Tensor& eigenvectors, Tensor& infos, const Tensor& input, bool compute_eigenvectors) { in linalg_eig_kernel()
222 …ensor& values, const Tensor& vectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in apply_lapack_eigh()
294 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_kernel()
|
| H A D | BatchLinearAlgebra.cpp | 2848 …o(const Tensor& input, Tensor& values, Tensor& vectors, Tensor& infos, bool compute_eigenvectors) { in linalg_eig_out_info()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebraLib.cpp | 1215 …ensor& values, const Tensor& vectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in apply_syevd()
1302 …ensor& values, const Tensor& vectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in apply_syevj()
1359 …ensor& values, const Tensor& vectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in apply_syevj_batched()
1418 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_cusolver_syevd()
1424 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_cusolver_syevj()
1430 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_cusolver_syevj_batched()
1436 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_cusolver()
|
| H A D | BatchLinearAlgebra.cpp | 1884 …ensor& values, const Tensor& vectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in apply_magma_eigh()
1961 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_magma()
1994 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_kernel()
2021 …alg_eig(Tensor& values, Tensor& vectors, Tensor& input, Tensor& infos, bool compute_eigenvectors) { in apply_linalg_eig()
2079 …eigenvalues, Tensor& eigenvectors, Tensor& infos, const Tensor& input, bool compute_eigenvectors) { in linalg_eig_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LinearAlgebraStubs.cpp | 88 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in lazy_linalg_eigh_kernel()
93 …eigenvalues, Tensor& eigenvectors, Tensor& infos, const Tensor& input, bool compute_eigenvectors) { in lazy_linalg_eig_kernel()
|