提交 c4cd50ee authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix bug.

上级 86244fa4
......@@ -684,7 +684,7 @@ GpuElemwise{mul}(lr, SparseBlockOuterSS) -> SparseBlockOuterSS(..., alpha=lr)
def local_merge_blocksparse_output(node):
if (isinstance(node.op, GpuElemwise) and
(node.op.scalar_op == scalar.sub or
node.scalar_op == scalar.add) and
node.op.scalar_op == scalar.add) and
node.nin == 2):
ger = grab_ger(node.inputs[0])
W = node.inputs[1]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论