Home
last modified time | relevance | path

Searched refs:cholesky_inverse_kernel_impl_cusolver (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebraLib.h69 Tensor& cholesky_inverse_kernel_impl_cusolver(Tensor &result, Tensor& infos, bool upper);
H A DBatchLinearAlgebra.cpp1440 return cholesky_inverse_kernel_impl_cusolver(result, infos, upper); in cholesky_inverse_kernel_impl()
1446 return cholesky_inverse_kernel_impl_cusolver(result, infos, upper); in cholesky_inverse_kernel_impl()
H A DBatchLinearAlgebraLib.cpp948 Tensor& cholesky_inverse_kernel_impl_cusolver(Tensor &result, Tensor& infos, bool upper) { in cholesky_inverse_kernel_impl_cusolver() function