| /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() argument
1219 …cusolverEigMode_t jobz = compute_eigenvectors ? CUSOLVER_EIG_MODE_VECTOR : CUSOLVER_EIG_MODE_NOVEC… in apply_syevd()
1302 …ensor& values, const Tensor& vectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in apply_syevj() argument
1306 …cusolverEigMode_t jobz = compute_eigenvectors ? CUSOLVER_EIG_MODE_VECTOR : CUSOLVER_EIG_MODE_NOVEC… in apply_syevj()
1359 …ensor& values, const Tensor& vectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in apply_syevj_batched() argument
1363 …cusolverEigMode_t jobz = compute_eigenvectors ? CUSOLVER_EIG_MODE_VECTOR : CUSOLVER_EIG_MODE_NOVEC… in apply_syevj_batched()
1418 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_cusolver_syevd() argument
1420 apply_syevd<scalar_t>(eigenvalues, eigenvectors, infos, upper, 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() argument
1426 apply_syevj<scalar_t>(eigenvalues, eigenvectors, infos, upper, compute_eigenvectors); in linalg_eigh_cusolver_syevj()
[all …]
|
| H A D | BatchLinearAlgebra.cpp | 1884 …ensor& values, const Tensor& vectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in apply_magma_eigh() argument
1897 magma_vec_t jobz = compute_eigenvectors ? MagmaVec : MagmaNoVec; in apply_magma_eigh()
1961 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_magma() argument
1973 eigenvalues_cpu, eigenvectors, infos_cpu, upper, compute_eigenvectors); in linalg_eigh_magma()
1983 if (compute_eigenvectors) { in linalg_eigh_magma()
1994 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_kernel() argument
1999 linalg_eigh_magma(eigenvalues, eigenvectors, infos, upper, compute_eigenvectors); in linalg_eigh_kernel()
2003 linalg_eigh_cusolver(eigenvalues, eigenvectors, infos, upper, compute_eigenvectors); in linalg_eigh_kernel()
2006 linalg_eigh_magma(eigenvalues, eigenvectors, infos, upper, compute_eigenvectors); in linalg_eigh_kernel()
2021 …alg_eig(Tensor& values, Tensor& vectors, Tensor& input, Tensor& infos, bool compute_eigenvectors) { in apply_linalg_eig() argument
[all …]
|
| H A D | BatchLinearAlgebraLib.h | 75 …genvalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors);
|
| /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() argument
90 linalg_eigh_stub(DeviceType::CUDA, eigenvalues, eigenvectors, infos, upper, compute_eigenvectors); in lazy_linalg_eigh_kernel()
93 …eigenvalues, Tensor& eigenvectors, Tensor& infos, const Tensor& input, bool compute_eigenvectors) { in lazy_linalg_eig_kernel() argument
95 linalg_eig_stub(DeviceType::CUDA, eigenvalues, eigenvectors, infos, input, compute_eigenvectors); in lazy_linalg_eig_kernel()
|
| /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() argument
153 char jobvr = compute_eigenvectors ? 'V' : 'N'; in apply_linalg_eig()
163 auto rvectors_data = compute_eigenvectors ? vectors.data_ptr<scalar_t>() : nullptr; in apply_linalg_eig()
165 int64_t ldvr = compute_eigenvectors ? lda : 1; in apply_linalg_eig()
188 …scalar_t* rvectors_working_ptr = compute_eigenvectors ? &rvectors_data[i * input_matrix_stride] : … in apply_linalg_eig()
197 …eigenvalues, Tensor& eigenvectors, Tensor& infos, const Tensor& input, bool compute_eigenvectors) { in linalg_eig_kernel() argument
208 …y_linalg_eig<scalar_t>(eigenvalues, eigenvectors, input_working_copy, infos, compute_eigenvectors); in linalg_eig_kernel()
222 …ensor& values, const Tensor& vectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in apply_lapack_eigh() argument
232 char jobz = compute_eigenvectors ? 'V' : 'N'; in apply_lapack_eigh()
294 …envalues, const Tensor& eigenvectors, const Tensor& infos, bool upper, bool compute_eigenvectors) { in linalg_eigh_kernel() argument
[all …]
|
| H A D | BatchLinearAlgebra.cpp | 2848 …o(const Tensor& input, Tensor& values, Tensor& vectors, Tensor& infos, bool compute_eigenvectors) { in linalg_eig_out_info() argument
2863 if (compute_eigenvectors) { in linalg_eig_out_info()
2874 if (vectors.numel() == 0 && compute_eigenvectors) { in linalg_eig_out_info()
2886 if (compute_eigenvectors) { in linalg_eig_out_info()
2912 if (vectors.is_complex() && compute_eigenvectors) { in linalg_eig_out_info()
2923 …ub(at::kCPU, real_imag_values, maybe_complex_vectors, infos, input.to(kCPU), compute_eigenvectors); in linalg_eig_out_info()
2925 …nput.device().type(), real_imag_values, maybe_complex_vectors, infos, input, compute_eigenvectors); in linalg_eig_out_info()
2938 if (compute_eigenvectors) { in linalg_eig_out_info()
2949 if (compute_eigenvectors) { in linalg_eig_out_info()
|