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):
509509 # Else, has to replace the s of this node with s of an SVD Op that compute_uv=False.
510510 # First, iterate to see if there is an SVD Op that can be reused.
511511 for cl , _ in fgraph .clients [x ]:
512- if cl == "output" :
513- continue
514512 if isinstance (cl .op , Blockwise ) and isinstance (cl .op .core_op , SVD ):
515513 if not cl .op .core_op .compute_uv :
516514 return {
@@ -529,8 +527,6 @@ def svd_uv_merge(fgraph, node):
529527 # We want rewrite if there is another one with compute_uv=True.
530528 # For this case, just reuse the `s` from the one with compute_uv=True.
531529 for cl , _ in fgraph .clients [x ]:
532- if cl == "output" :
533- continue
534530 if isinstance (cl .op , Blockwise ) and isinstance (cl .op .core_op , SVD ):
535531 if cl .op .core_op .compute_uv and (
536532 len (fgraph .clients [cl .outputs [0 ]]) > 0
You can’t perform that action at this time.
0 commit comments