Home
last modified time | relevance | path

Searched refs:Cusolver (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/aten/src/ATen/
DLinalgBackend.h10 enum class LinalgBackend : int8_t { Default, Cusolver, Magma }; enumerator
16 case LinalgBackend::Cusolver: in LinalgBackendToString()
DContext.cpp253 TORCH_CHECK((b != at::LinalgBackend::Cusolver) || hasCuSOLVER(), in setLinalgPreferredBackend()
DContext.h385 ? at::LinalgBackend::Cusolver
/external/pytorch/torch/backends/cuda/
D__init__.py148 "cusolver": torch._C._LinalgBackend.Cusolver,
/external/pytorch/aten/src/ATen/native/cuda/linalg/
DBatchLinearAlgebra.cpp1110 case at::LinalgBackend::Cusolver: in ldl_factor_kernel()
1244 case at::LinalgBackend::Cusolver: in _cholesky_solve_helper_cuda()
1358 case at::LinalgBackend::Cusolver: in cholesky_kernel()
1439 case at::LinalgBackend::Cusolver: in cholesky_inverse_kernel_impl()
1634 if (preferred_backend == at::LinalgBackend::Cusolver) { in lu_factor()
1872 case at::LinalgBackend::Cusolver: in geqrf_kernel()
2001 case at::LinalgBackend::Cusolver: in linalg_eigh_kernel()
2492 if (preferred_backend == at::LinalgBackend::Cusolver) { in lu_solve_kernel()
2719 case at::LinalgBackend::Cusolver: in gels_looped()
/external/pytorch/torch/csrc/
DModule.cpp2055 .value("Cusolver", at::LinalgBackend::Cusolver) in initModule()
/external/pytorch/torch/_C/
D__init__.pyi.in1252 Cusolver: _LinalgBackend