提交 dbf5f38e authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Refactor reshape + dimshuffle rewrites

上级 02545ed5
...@@ -2800,16 +2800,6 @@ def _check_chain(r, chain): ...@@ -2800,16 +2800,6 @@ def _check_chain(r, chain):
return r is not None return r is not None
def check_chain(r, *chain):
"""
WRITEME
"""
if isinstance(r, Apply):
r = r.outputs[0]
return _check_chain(r, reduce(list.__iadd__, ([x, 0] for x in chain)))
def pre_greedy_node_rewriter( def pre_greedy_node_rewriter(
fgraph: FunctionGraph, rewrites: Sequence[NodeRewriter], out: Variable fgraph: FunctionGraph, rewrites: Sequence[NodeRewriter], out: Variable
) -> Variable: ) -> Variable:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论