Skip to content

Commit dbf5f38

Browse files
committed
Refactor reshape + dimshuffle rewrites
1 parent 02545ed commit dbf5f38

File tree

2 files changed

+150
-171
lines changed

2 files changed

+150
-171
lines changed

pytensor/graph/rewriting/basic.py

-10
Original file line numberDiff line numberDiff line change
@@ -2800,16 +2800,6 @@ def _check_chain(r, chain):
28002800
return r is not None
28012801

28022802

2803-
def check_chain(r, *chain):
2804-
"""
2805-
WRITEME
2806-
2807-
"""
2808-
if isinstance(r, Apply):
2809-
r = r.outputs[0]
2810-
return _check_chain(r, reduce(list.__iadd__, ([x, 0] for x in chain)))
2811-
2812-
28132803
def pre_greedy_node_rewriter(
28142804
fgraph: FunctionGraph, rewrites: Sequence[NodeRewriter], out: Variable
28152805
) -> Variable:

0 commit comments

Comments
 (0)