diff --git a/imperative/python/megengine/utils/network_node.py b/imperative/python/megengine/utils/network_node.py index 5e1f1af339729d377eb139e029bf94c612b6a3be..155b057965a8513e5eb3441005135b47a5b03171 100644 --- a/imperative/python/megengine/utils/network_node.py +++ b/imperative/python/megengine/utils/network_node.py @@ -99,6 +99,9 @@ class OpNode(NetworkNode): def add_out_var(self, x): self.outputs.append(x) + def __repr__(self): + return "%s{%s}" % (self.name, self.type) + def str_to_mge_class(classname): # TODO: use megbrain C++ RTTI to replace type string @@ -230,11 +233,17 @@ class Elemwise(OpNode): type = "Elemwise" opdef = builtin.Elemwise + def __repr__(self): + return "%s{Elemwise:%s}" % (self.name, self.params["mode"]) + class ElemwiseMultiType(OpNode): type = "ElemwiseMultiType" opdef = builtin.ElemwiseMultiType + def __repr__(self): + return "%s{ElemwiseMultiType:%s}" % (self.name, self.params["mode"]) + @classmethod def load(cls, opr): obj = super(ElemwiseMultiType, cls).load(opr)