提交 f52cd452 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

fixed desc in elemwise2

上级 b2fd4e48
...@@ -157,7 +157,7 @@ class Broadcast(Op, Destroyer): ...@@ -157,7 +157,7 @@ class Broadcast(Op, Destroyer):
return Broadcast(self.scalar_opclass, new_inputs, self.inplace_pattern) return Broadcast(self.scalar_opclass, new_inputs, self.inplace_pattern)
def desc(self): def desc(self):
return (self.__class__, self.scalar_opclass, tuple(self.inplace_pattern.items())) return (Broadcast, self.scalar_opclass, tuple(self.inplace_pattern.items()))
def destroy_map(self): def destroy_map(self):
ret = {} ret = {}
...@@ -311,6 +311,9 @@ def make_broadcast(scalar_opclass, inplace_pattern = {}, name = None): ...@@ -311,6 +311,9 @@ def make_broadcast(scalar_opclass, inplace_pattern = {}, name = None):
Broadcast.__init__(self, scalar_opclass, inputs, inplace_pattern) Broadcast.__init__(self, scalar_opclass, inputs, inplace_pattern)
def clone_with_new_inputs(self, *new_inputs): def clone_with_new_inputs(self, *new_inputs):
return New(*new_inputs) return New(*new_inputs)
@classmethod
def desc(cls):
return (Broadcast, scalar_opclass, tuple(inplace_pattern.items()))
if name is not None: if name is not None:
New.__name__ = name New.__name__ = name
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论