Searched defs:compute_pivots (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebra.cpp | 1476 …looped_magma(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pivots) { in apply_lu_factor_looped_magma()
1530 …atched_magma(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pivots) { in apply_lu_factor_batched_magma()
1589 …looped_magma(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pivots) { in lu_factor_looped_magma()
1595 …atched_magma(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pivots) { in lu_factor_batched_magma()
1601 …id lu_factor(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pivots) { in lu_factor()
1607 …_size](const Tensor& input, const Tensor& pivots, const Tensor& infos, const bool compute_pivots) { in lu_factor()
1617 …_size, m, n](const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pivots) { in lu_factor()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebraKernel.cpp | 898 …ly_lu_factor(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pivots) { in apply_lu_factor()
943 …actor_kernel(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pivots) { in lu_factor_kernel()
|
| H A D | BatchLinearAlgebra.cpp | 1989 std::tuple<Tensor, Tensor, Tensor> _lu_with_info(const Tensor& self, bool compute_pivots, bool) { in _lu_with_info() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LinearAlgebraStubs.cpp | 63 …zy_lu_factor(const Tensor& input, const Tensor& pivots, const Tensor& infos, bool compute_pivots) { in lazy_lu_factor()
|