提交 ca58eeb6 authored 作者: Frederic Bastien's avatar Frederic Bastien

small speed up

上级 df0dd2c1
......@@ -482,7 +482,7 @@ class IgnorePrinter:
class DefaultPrinter:
def __init__(self):
pass
self.leaf_printer = LeafPrinter()
def process(self, output, pstate):
if output in pstate.memo:
......@@ -490,7 +490,7 @@ class DefaultPrinter:
pprinter = pstate.pprinter
node = output.owner
if node is None:
return LeafPrinter().process(output, pstate)
return self.leaf_printer.process(output, pstate)
r = "%s(%s)" % (str(node.op), ", ".join(
[pprinter.process(input, pstate.clone(precedence=-1000))
for input in node.inputs]))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论