提交 48240dcd authored 作者: James Bergstra's avatar James Bergstra

do constant folding in specialization pass too

上级 0e5affcd
...@@ -1041,6 +1041,7 @@ def constant_folding(node): ...@@ -1041,6 +1041,7 @@ def constant_folding(node):
return [gof.Constant(output.type, s[0]) for s, output in zip(storage, node.outputs)] return [gof.Constant(output.type, s[0]) for s, output in zip(storage, node.outputs)]
register_canonicalize(constant_folding) register_canonicalize(constant_folding)
register_specialize(constant_folding)
inplace_matrix_transpose = T.DimShuffle([False,False], [1,0], inplace=True) inplace_matrix_transpose = T.DimShuffle([False,False], [1,0], inplace=True)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论