提交 1597f7c5 authored 作者: Frederic Bastien's avatar Frederic Bastien

print the full scalar composite op inside GpuElemwise when they are done inplace.

上级 6c99aa83
......@@ -124,8 +124,9 @@ class GpuElemwise(Op):
if self.inplace_pattern:
items = self.inplace_pattern.items()
items.sort()
return "GpuElemwise{%s}%s" % (self.scalar_op.__class__.__name__, str(items))
#return "GpuElemwise{%s}" % (self.scalar_op.__class__.__name__)
# We need to print the scalar_op, not only the its class name
# to have the full definition of composite op.
return "GpuElemwise{%s}%s" % (self.scalar_op, str(items))
return "GpuElemwise{%s}" % (self.scalar_op)
def __repr__(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论