提交 f06637e0 authored 作者: sentient07's avatar sentient07

Added __props__ at few more places

上级 774c4a91
...@@ -72,6 +72,8 @@ class OpFromGraph(gof.Op): ...@@ -72,6 +72,8 @@ class OpFromGraph(gof.Op):
""" """
__props__ = ("inputs", "outputs")
def __init__(self, inputs, outputs, **kwargs): def __init__(self, inputs, outputs, **kwargs):
if not isinstance(outputs, list): if not isinstance(outputs, list):
raise TypeError('outputs must be list', outputs) raise TypeError('outputs must be list', outputs)
......
...@@ -518,6 +518,8 @@ class FromFunctionOp(gof.Op): ...@@ -518,6 +518,8 @@ class FromFunctionOp(gof.Op):
""" """
__props__ = ("fn", "itypes", "otypes", "infer_shape")
def __init__(self, fn, itypes, otypes, infer_shape): def __init__(self, fn, itypes, otypes, infer_shape):
self.__fn = fn self.__fn = fn
self.itypes = itypes self.itypes = itypes
......
...@@ -48,6 +48,10 @@ def MyConstant(data): ...@@ -48,6 +48,10 @@ def MyConstant(data):
class MyOp(Op): class MyOp(Op):
__props__ = ("nin", "name", "vmap", "dmap", "nout",
"destroyhandler_tolerate_same",
"destroyhandler_tolerate_aliased")
def __init__(self, nin, name, vmap=None, dmap=None, nout=1, def __init__(self, nin, name, vmap=None, dmap=None, nout=1,
destroyhandler_tolerate_same=None, destroyhandler_tolerate_same=None,
destroyhandler_tolerate_aliased=None): destroyhandler_tolerate_aliased=None):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论