提交 b2e969c3 authored 作者: Frederic Bastien's avatar Frederic Bastien

Disable fusion for elemwise with multiple output.

上级 c0b24762
...@@ -7001,6 +7001,10 @@ def local_elemwise_fusion_op(OP, max_input_fct=lambda node: 32, ...@@ -7001,6 +7001,10 @@ def local_elemwise_fusion_op(OP, max_input_fct=lambda node: 32,
if type(node.op) is not OP: if type(node.op) is not OP:
return False return False
if len(node.outputs) > 1:
# We don't support the fusion for node with multiple outputs.
return
inputs = [] # inputs of the new Elemwise op. inputs = [] # inputs of the new Elemwise op.
s_inputs = [] # inputs of the new scalar op used by the Composite. s_inputs = [] # inputs of the new scalar op used by the Composite.
# Inputs of the new scalar op that represents the current node. # Inputs of the new scalar op that represents the current node.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论