Searched refs:linalg_solve_triangular (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 1933 auto dL = at::linalg_solve_triangular(L_, dA, /*upper=*/false, /*left=*/true); in cholesky_jvp()
1934 dL = at::linalg_solve_triangular(L_.mH(), dL, /*upper=*/true, /*left=*/false); in cholesky_jvp()
1963 gA = at::linalg_solve_triangular(L_.mH(), gA, /*upper=*/true, /*left=*/true); in cholesky_backward()
1964 gA = at::linalg_solve_triangular(L_, gA, /*upper=*/false, /*left=*/false); in cholesky_backward()
3957 at::linalg_solve_triangular(R, dA, /*upper=*/true, /*left=*/false); in linalg_qr_jvp()
3983 auto QHdA1Rinv = at::linalg_solve_triangular( in linalg_qr_jvp()
4069 gA = at::linalg_solve_triangular( in linalg_qr_backward()
4081 gA = at::linalg_solve_triangular( in linalg_qr_backward()
4432 return at::linalg_solve_triangular(A, X_t, upper, left, unitriangular); in linalg_solve_triangular_forward_AD()
4480 at::linalg_solve_triangular(A_H, grad, !upper, left, unitriangular); in linalg_solve_triangular_backward()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/linalg/ |
| H A D | __init__.py | 2194 solve_triangular = _add_docstr(_linalg.linalg_solve_triangular, r"""
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 892 aten::linalg_solve_triangular
893 aten::linalg_solve_triangular.out
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | linalg.h | 367 return torch::linalg_solve_triangular( in solve_triangular()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesLinearAlgebra.cpp | 740 LINALG_CHECK_MATRIX_BINARY_ONE_OUT(linalg_solve_triangular, linalg.solve_triangular);
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | generated_ops.cpp | 2817 aten::linalg_solve_triangular,
2829 p_node->Output(0) = at::native::linalg_solve_triangular( in __anon5d9c3eb910e02()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 1458 @register_meta([aten.linalg_solve_triangular.default, aten.linalg_solve_triangular.out])
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebra.cpp | 4000 Tensor linalg_solve_triangular( in linalg_solve_triangular() function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 2327 make_fallback(aten.linalg_solve_triangular)
|