提交 c6bbc478 编写于 作者: M Megvii Engine Team

feat(imperative/utils): add __repr__ for network node

GitOrigin-RevId: fe5b59749fb98fc9fdca61421ff6e6d657472fe4
上级 04b1a45a
...@@ -99,6 +99,9 @@ class OpNode(NetworkNode): ...@@ -99,6 +99,9 @@ class OpNode(NetworkNode):
def add_out_var(self, x): def add_out_var(self, x):
self.outputs.append(x) self.outputs.append(x)
def __repr__(self):
return "%s{%s}" % (self.name, self.type)
def str_to_mge_class(classname): def str_to_mge_class(classname):
# TODO: use megbrain C++ RTTI to replace type string # TODO: use megbrain C++ RTTI to replace type string
...@@ -230,11 +233,17 @@ class Elemwise(OpNode): ...@@ -230,11 +233,17 @@ class Elemwise(OpNode):
type = "Elemwise" type = "Elemwise"
opdef = builtin.Elemwise opdef = builtin.Elemwise
def __repr__(self):
return "%s{Elemwise:%s}" % (self.name, self.params["mode"])
class ElemwiseMultiType(OpNode): class ElemwiseMultiType(OpNode):
type = "ElemwiseMultiType" type = "ElemwiseMultiType"
opdef = builtin.ElemwiseMultiType opdef = builtin.ElemwiseMultiType
def __repr__(self):
return "%s{ElemwiseMultiType:%s}" % (self.name, self.params["mode"])
@classmethod @classmethod
def load(cls, opr): def load(cls, opr):
obj = super(ElemwiseMultiType, cls).load(opr) obj = super(ElemwiseMultiType, cls).load(opr)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册