We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1d46aef commit 455707eCopy full SHA for 455707e
pytensor/tensor/rewriting/linalg.py
@@ -50,7 +50,7 @@ def inv_as_solve(fgraph, node):
50
@register_stabilize
51
@register_canonicalize
52
@node_rewriter([Solve])
53
-def tag_solve_triangular(fgraph, node):
+def generic_solve_to_solve_triangular(fgraph, node):
54
"""
55
If any solve() is applied to the output of a cholesky op, then
56
replace it with a triangular solve.
tests/tensor/rewriting/test_linalg.py
@@ -82,7 +82,7 @@ def test_transinv_to_invtrans():
82
assert node.inputs[0].name == "X"
83
84
85
-def test_tag_solve_triangular():
+def test_generic_solve_to_solve_triangular():
86
cholesky_lower = Cholesky(lower=True)
87
cholesky_upper = Cholesky(lower=False)
88
A = matrix("A")
0 commit comments