| /external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| D | BatchLinearAlgebraLibBlas.cpp | 161 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in apply_triangular_solve() argument
165 cublasDiagType_t diag = unitriangular ? CUBLAS_DIAG_UNIT : CUBLAS_DIAG_NON_UNIT; in apply_triangular_solve()
188 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in triangular_solve_cublas() argument
190 apply_triangular_solve<scalar_t>(A, B, left, upper, transpose, 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() argument
199 cublasDiagType_t diag = unitriangular ? CUBLAS_DIAG_UNIT : CUBLAS_DIAG_NON_UNIT; 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() argument
230 triangular_solve_batched_cublas(A, b, left, upper, transpose, unitriangular); in triangular_solve_batched_cublas()
236 apply_triangular_solve_batched<scalar_t>(A, B, left, upper, transpose, unitriangular); in triangular_solve_batched_cublas()
|
| D | BatchLinearAlgebraLib.h | 43 …st Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular);
44 …st Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular);
|
| D | BatchLinearAlgebra.cpp | 1678 …t Tensor& A, const Tensor& b, bool left, bool upper, TransposeType transpose, bool unitriangular) { in apply_triangular_solve_batched_magma() argument
1685 magma_diag_t diag = unitriangular ? MagmaUnit : MagmaNonUnit; 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() argument
1745 apply_triangular_solve_batched_magma<scalar_t>(A, B, left, upper, transpose, 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() argument
1752 triangular_solve_cublas(A, B, left, upper, transpose, unitriangular); in triangular_solve_kernel()
1755 triangular_solve_batched_cublas(A, B, left, upper, transpose, unitriangular); in triangular_solve_kernel()
1759 triangular_solve_batched_cublas(A, B, left, upper, transpose, unitriangular); in triangular_solve_kernel()
1761 triangular_solve_batched_magma(A, B, left, upper, transpose, unitriangular); in triangular_solve_kernel()
|
| /external/pytorch/aten/src/ATen/native/mkl/ |
| D | SparseBlasImpl.h | 37 bool unitriangular);
|
| D | SparseBlasImpl.cpp | 617 bool unitriangular) { in triangular_solve_out_sparse_csr() argument
647 const auto A = unitriangular ? materialize_diagonal_indices(A_) : A_; in triangular_solve_out_sparse_csr()
661 descrA.diag = unitriangular ? SPARSE_DIAG_UNIT : SPARSE_DIAG_NON_UNIT; in triangular_solve_out_sparse_csr()
|
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseBlasImpl.cpp | 135 bool unitriangular) { in block_sparse_triangular_solve_vec() argument
172 auto desc = at::cuda::sparse::CuSparseMatDescriptor(upper, unitriangular); in block_sparse_triangular_solve_vec()
222 if (!unitriangular) { in block_sparse_triangular_solve_vec()
263 bool unitriangular) { in block_sparse_triangular_solve_mat() argument
308 auto desc = at::cuda::sparse::CuSparseMatDescriptor(upper, unitriangular); in block_sparse_triangular_solve_mat()
359 if (!unitriangular) { in block_sparse_triangular_solve_mat()
1262 bool unitriangular) { in triangular_solve_out_sparse_csr() argument
1270 return block_sparse_triangular_solve_vec(A, B, X, upper, transpose, unitriangular); in triangular_solve_out_sparse_csr()
1272 return block_sparse_triangular_solve_mat(A, B, X, upper, transpose, unitriangular); in triangular_solve_out_sparse_csr()
1292 descA.set_mat_diag_type(unitriangular); in triangular_solve_out_sparse_csr()
|
| D | SparseBlasImpl.h | 37 bool unitriangular);
|
| D | SparseBlas.cpp | 280 bool unitriangular, in triangular_solve_out_sparse_csr_cuda() argument
283 sparse::impl::cuda::triangular_solve_out_sparse_csr(A, B, X, upper, transpose, unitriangular); in triangular_solve_out_sparse_csr_cuda()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | LinearAlgebraStubs.cpp | 68 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in lazy_triangular_solve_kernel() argument
70 triangular_solve_stub(DeviceType::CUDA, A, B, left, upper, transpose, unitriangular); in lazy_triangular_solve_kernel()
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseBlasImpl.h | 42 bool unitriangular);
|
| D | SparseBlas.cpp | 101 bool unitriangular, in triangular_solve_out_sparse_csr_cpu() argument
104 sparse::impl::cpu::triangular_solve_out_sparse_csr(A, B, X, upper, transpose, unitriangular); in triangular_solve_out_sparse_csr_cpu()
|
| D | SparseBlasImpl.cpp | 461 bool unitriangular) { in triangular_solve_out_sparse_csr() argument
469 sparse::impl::mkl::triangular_solve_out_sparse_csr(A, B, X, upper, transpose, unitriangular); in triangular_solve_out_sparse_csr()
|
| /external/pytorch/torch/csrc/api/include/torch/ |
| D | linalg.h | 366 bool unitriangular) { in solve_triangular() argument
368 input, other, upper, left, unitriangular); in solve_triangular()
377 bool unitriangular) { in solve_triangular_out() argument
379 result, input, other, upper, left, unitriangular); in solve_triangular_out()
958 bool unitriangular) { in solve_triangular() argument
959 return detail::solve_triangular(input, other, upper, left, unitriangular); in solve_triangular()
968 bool unitriangular) { in solve_triangular_out() argument
970 result, input, other, upper, left, unitriangular); in solve_triangular_out()
|
| /external/pytorch/aten/src/ATen/native/mps/operations/ |
| D | LinearAlgebra.mm | 712 bool unitriangular,
757 … unit:unitriangular
979 bool unitriangular,
981 … mps::linalg_solve_triangular_mps_impl(A, B, upper, /*transpose=*/false, left, unitriangular, out);
984 …solve_triangular_mps(const Tensor& A, const Tensor& B, bool upper, bool left, bool unitriangular) {
986 mps::linalg_solve_triangular_mps_impl(A, B, upper, /*transpose=*/false, left, unitriangular, out);
995 bool unitriangular,
1000 …mps::linalg_solve_triangular_mps_impl(A, self, upper, transpose, /*left=*/true, unitriangular, out…
|
| /external/pytorch/test/ |
| D | test_sparse_csr.py | 1768 def run_test(a, b, upper, transpose, unitriangular, op_out): argument
1769 if unitriangular and self.device_type == 'cpu':
1777 …actual = torch.triangular_solve(b, a, upper=upper, unitriangular=unitriangular, transpose=transpos…
1799 unitriangular=unitriangular)
1813 … upper=upper, unitriangular=unitriangular, transpose=transpose, out=(out, actual_A_clone)
1832 … for (upper, unitriangular, transpose, op_out) in itertools.product([True, False], repeat=4):
1833 run_test(a, b, upper, unitriangular, transpose, op_out)
2360 def run_test(n, k, upper, unitriangular, transpose, zero): argument
2361 if not unitriangular:
2383 …expected = torch.triangular_solve(B, A, upper=upper, unitriangular=unitriangular, transpose=transp…
[all …]
|
| D | test_linalg.py | 4126 X = torch.linalg.solve_triangular(A, B, upper=upper, left=left, unitriangular=uni)
4135 torch.linalg.solve_triangular(A, B, upper=upper, left=left, unitriangular=uni, out=out)
4198 X = torch.linalg.solve_triangular(A, B, upper=upper, left=left, unitriangular=uni)
4206 def triangular_solve_test_helper(self, A_dims, b_dims, upper, unitriangular, argument
4214 if unitriangular:
4227 for k, n, (upper, unitriangular, transpose) in itertools.product(ks, ns,
4230 unitriangular, device, dtype)
4231 … x = torch.triangular_solve(b, A, upper=upper, unitriangular=unitriangular, transpose=transpose)[0]
4243 def triangular_solve_batch_helper(A_dims, b_dims, upper, unitriangular, transpose): argument
4245 unitriangular, device, dtype)
[all …]
|
| /external/pytorch/aten/src/ATen/native/ |
| D | BatchLinearAlgebraKernel.cpp | 716 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in apply_triangular_solve() argument
724 char diag = unitriangular ? 'U' : 'N'; in apply_triangular_solve()
747 …t Tensor& A, const Tensor& B, bool left, bool upper, TransposeType transpose, bool unitriangular) { in triangular_solve_kernel() argument
749 apply_triangular_solve<scalar_t>(A, B, left, upper, transpose, unitriangular); in triangular_solve_kernel()
|
| D | BatchLinearAlgebra.cpp | 476 …gular_solve)(const Tensor& self, const Tensor& A, bool upper, bool transpose, bool unitriangular) { in TORCH_META_FUNC()
2189 bool upper, bool transpose, bool unitriangular) { in triangular_solve_out_impl() argument
2235 …t=*/true, upper, transpose ? TransposeType::Transpose : TransposeType::NoTranspose, unitriangular); in triangular_solve_out_impl()
2238 …const Tensor& self, const Tensor& A, bool upper, bool transpose, bool unitriangular, const Tensor&… in TORCH_IMPL_FUNC()
2248 …ve_out_impl(result_tmp, clone_A_tmp, A_broadcast, self_broadcast, upper, transpose, unitriangular); in TORCH_IMPL_FUNC()
2253 …ular_solve_out_impl(result, clone_A, A_broadcast, self_broadcast, upper, transpose, unitriangular); in TORCH_IMPL_FUNC()
3818 bool unitriangular, in linalg_solve_triangular_out() argument
3988 /*unitriangular=*/unitriangular); in linalg_solve_triangular_out()
4005 bool unitriangular) { in linalg_solve_triangular() argument
4007 linalg_solve_triangular_out(A, B, upper, left, unitriangular, out); in linalg_solve_triangular()
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.h | 656 const bool unitriangular,
665 const bool unitriangular);
673 const bool unitriangular);
680 const bool unitriangular,
|
| D | FunctionsManual.cpp | 4366 const bool unitriangular, in triangular_solve_backward() argument
4373 grad_x.triangular_solve(a.conj(), upper, !transpose, unitriangular)); in triangular_solve_backward()
4378 grad_a = grad_a.triu((int)unitriangular); in triangular_solve_backward()
4380 grad_a = grad_a.tril(-((int)unitriangular)); in triangular_solve_backward()
4404 const bool unitriangular) { in triangular_solve_jvp() argument
4408 dB - dA_contrib, A, upper, transpose, unitriangular)); in triangular_solve_jvp()
4423 const bool unitriangular) { in linalg_solve_triangular_forward_AD() argument
4428 const Tensor proj_A_t = upper ? A_t.triu(static_cast<int>(unitriangular)) in linalg_solve_triangular_forward_AD()
4429 : A_t.tril(-static_cast<int>(unitriangular)); in linalg_solve_triangular_forward_AD()
4432 return at::linalg_solve_triangular(A, X_t, upper, left, unitriangular); in linalg_solve_triangular_forward_AD()
[all …]
|
| /external/pytorch/torch/csrc/inductor/aoti_torch/generated/ |
| D | c_shim_cpu.h | 126 …e self, AtenTensorHandle A, int32_t upper, int32_t transpose, int32_t unitriangular, AtenTensorHan…
|
| D | c_shim_cuda.h | 135 …e self, AtenTensorHandle A, int32_t upper, int32_t transpose, int32_t unitriangular, AtenTensorHan…
|
| /external/pytorch/torch/ |
| D | overrides.py | 1231 … torch.triangular_solve: lambda input, A, upper=True, transpose=False, unitriangular=False: -1,
1232 torch.linalg.solve_triangular: lambda input, B, upper, left=True, unitriangular=False: -1,
|
| D | _meta_registrations.py | 1465 unitriangular: bool = False,
1491 unitriangular: bool = False,
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | generated_ops.cpp | 2827 const auto unitriangular = p_node->Input(4).toBool(); in __anon7ebb593910e02() local
2830 self, B, upper, left, unitriangular); in __anon7ebb593910e02()
2836 self, B, upper, left, unitriangular, out); in __anon7ebb593910e02()
|