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

better str representation of Composite op.

上级 772395e3
......@@ -1211,7 +1211,8 @@ class Composite(ScalarOp):
else:
return "%s{%s}" % (self.__class__.__name__, ", ".join(
"%s=%s" % (k, v) for k, v in self.__dict__.items()
if k not in ["name","env","_c_code"] ))
if k not in ["env","_c_code", "_cmodule_key", "_impls",
"_hashval"] ))
def __init__(self, inputs, outputs):
env = Env(*gof.graph.clone(inputs, outputs))
......@@ -1234,6 +1235,9 @@ class Composite(ScalarOp):
else:
raise ValueError("All orphans in the env to Composite must be Constant instances.")
if not hasattr(self,"name"):
self.name="".join([n.op.__class__.__name__ if not hasattr(n,"name") else n.name for n in env.nodes])
_c_code = "{\n"
i = 0
j = 0
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论