File tree Expand file tree Collapse file tree 1 file changed +0
-4
lines changed
pytensor/tensor/rewriting Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Original file line number Diff line number Diff line change @@ -509,8 +509,6 @@ def svd_uv_merge(fgraph, node):
509
509
# Else, has to replace the s of this node with s of an SVD Op that compute_uv=False.
510
510
# First, iterate to see if there is an SVD Op that can be reused.
511
511
for cl , _ in fgraph .clients [x ]:
512
- if cl == "output" :
513
- continue
514
512
if isinstance (cl .op , Blockwise ) and isinstance (cl .op .core_op , SVD ):
515
513
if not cl .op .core_op .compute_uv :
516
514
return {
@@ -529,8 +527,6 @@ def svd_uv_merge(fgraph, node):
529
527
# We want rewrite if there is another one with compute_uv=True.
530
528
# For this case, just reuse the `s` from the one with compute_uv=True.
531
529
for cl , _ in fgraph .clients [x ]:
532
- if cl == "output" :
533
- continue
534
530
if isinstance (cl .op , Blockwise ) and isinstance (cl .op .core_op , SVD ):
535
531
if cl .op .core_op .compute_uv and (
536
532
len (fgraph .clients [cl .outputs [0 ]]) > 0
You can’t perform that action at this time.
0 commit comments