| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebraLibBlas.cpp | 161 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in apply_triangular_solve()
188 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in triangular_solve_cublas()
195 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in apply_triangular_solve_batched()
220 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in triangular_solve_batched_cublas()
|
| H A D | BatchLinearAlgebra.cpp | 1678 …t Tensor& A, const Tensor& b, bool left, bool upper, TransposeType transpose, bool unitriangular) { in apply_triangular_solve_batched_magma()
1743 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in triangular_solve_batched_magma()
1749 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in triangular_solve_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseBlasImpl.cpp | 135 bool unitriangular) { in block_sparse_triangular_solve_vec()
263 bool unitriangular) { in block_sparse_triangular_solve_mat()
1262 bool unitriangular) { in triangular_solve_out_sparse_csr()
|
| H A D | SparseBlas.cpp | 280 bool unitriangular, in triangular_solve_out_sparse_csr_cuda()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | linalg.h | 366 bool unitriangular) { in solve_triangular()
377 bool unitriangular) { in solve_triangular_out()
958 bool unitriangular) { in solve_triangular()
968 bool unitriangular) { in solve_triangular_out()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebraKernel.cpp | 716 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in apply_triangular_solve()
747 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in triangular_solve_kernel()
|
| H A D | BatchLinearAlgebra.cpp | 2189 bool upper, bool transpose, bool unitriangular) { in triangular_solve_out_impl()
3818 bool unitriangular, in linalg_solve_triangular_out()
4005 bool unitriangular) { in linalg_solve_triangular()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseBlas.cpp | 101 bool unitriangular, in triangular_solve_out_sparse_csr_cpu()
|
| H A D | SparseBlasImpl.cpp | 461 bool unitriangular) { in triangular_solve_out_sparse_csr()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/ |
| H A D | SparseBlasImpl.cpp | 617 bool unitriangular) { in triangular_solve_out_sparse_csr()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LinearAlgebraStubs.cpp | 68 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in lazy_triangular_solve_kernel()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_linalg.py | 4206 def triangular_solve_test_helper(self, A_dims, b_dims, upper, unitriangular, argument
4243 def triangular_solve_batch_helper(A_dims, b_dims, upper, unitriangular, transpose): argument
4262 def triangular_solve_zero_batch_helper(A_dims, b_dims, upper, unitriangular, transpose): argument
4339 def run_test(A_dims, b_dims, device, upper, transpose, unitriangular): argument
|
| H A D | test_sparse_csr.py | 1768 def run_test(a, b, upper, transpose, unitriangular, op_out): argument
2360 def run_test(n, k, upper, unitriangular, transpose, zero): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 4366 const bool unitriangular, in triangular_solve_backward()
4404 const bool unitriangular) { in triangular_solve_jvp()
4423 const bool unitriangular) { in linalg_solve_triangular_forward_AD()
4441 const bool unitriangular, in linalg_solve_triangular_backward()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | generated_ops.cpp | 2827 const auto unitriangular = p_node->Input(4).toBool(); in __anon5d9c3eb910e02() local
|