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

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

GitOrigin-RevId: fe5b59749fb98fc9fdca61421ff6e6d657472fe4
上级 04b1a45a
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册