Searched refs:linalgPreferredBackend (Results 1 – 6 of 6) sorted by relevance
1093 at::globalContext().linalgPreferredBackend(), in ldl_factor_magma()1108 auto preferred_backend = at::globalContext().linalgPreferredBackend(); in ldl_factor_kernel()1242 auto preferred_backend = at::globalContext().linalgPreferredBackend(); in _cholesky_solve_helper_cuda()1356 auto preferred_backend = at::globalContext().linalgPreferredBackend(); in cholesky_kernel()1437 auto preferred_backend = at::globalContext().linalgPreferredBackend(); in cholesky_inverse_kernel_impl()1615 const auto preferred_backend = at::globalContext().linalgPreferredBackend(); in lu_factor()1859 auto preferred_backend = at::globalContext().linalgPreferredBackend(); in geqrf_kernel()1996 auto preferred_backend = at::globalContext().linalgPreferredBackend(); in linalg_eigh_kernel()2220 bool use_magma = at::globalContext().linalgPreferredBackend() == at::LinalgBackend::Magma; in svd_kernel()2490 auto preferred_backend = at::globalContext().linalgPreferredBackend(); in lu_solve_kernel()[all …]
219 at::globalContext().linalgPreferredBackend(), in ldl_factor_cusolver()
247 at::LinalgBackend Context::linalgPreferredBackend() const { in linalgPreferredBackend() function in at::Context
217 at::LinalgBackend linalgPreferredBackend() const;
414 && at::globalContext().linalgPreferredBackend() != at::LinalgBackend::Magma; in svd_uses_cusolver()
2062 return at::globalContext().linalgPreferredBackend(); in initModule()