提交 146a0a84 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Remove restrictive check in `local_blockwise_alloc`

上级 b79d2329
......@@ -188,6 +188,5 @@ def local_blockwise_alloc(fgraph, node):
)
for new_out in new_outs
]
assert new_outs[0].type.broadcastable == old_out_type.broadcastable
copy_stack_trace(node.outputs, new_outs)
return new_outs
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论