Home
last modified time | relevance | path

Searched defs:self_working_copy (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebraLib.cpp719 inline static void apply_cholesky_cusolver_potrf_looped(const Tensor& self_working_copy, bool upper… in apply_cholesky_cusolver_potrf_looped()
788 inline static void apply_cholesky_cusolver_potrfBatched(const Tensor& self_working_copy, bool upper… in apply_cholesky_cusolver_potrfBatched()
823 inline static void apply_cholesky_cusolver_potrs(Tensor& self_working_copy, const Tensor& A_column_… in apply_cholesky_cusolver_potrs()
879 inline static void apply_cholesky_cusolver_potrsBatched(Tensor& self_working_copy, const Tensor& A_… in apply_cholesky_cusolver_potrsBatched()
910 at::Tensor self_working_copy = cloneBatchedColumnMajor(self); in _cholesky_solve_helper_cuda_cusolver() local
H A DBatchLinearAlgebra.cpp1229 auto self_working_copy = cloneBatchedColumnMajor(self); in _cholesky_solve_helper_cuda_magma() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBatchLinearAlgebra.cpp1656 auto self_working_copy = cloneBatchedColumnMajor(self); in _cholesky_solve_helper_cpu() local