提交 bedebb5c 编写于 作者: C channingss

change sum op to elementwise_add op (for paddlelite)

上级 c7d1f2a8
...@@ -640,13 +640,13 @@ class ONNXOpMapper(OpMapper): ...@@ -640,13 +640,13 @@ class ONNXOpMapper(OpMapper):
param_attr=attr) param_attr=attr)
def Sum(self, node): def Sum(self, node):
val_inps = [val for val in node.layer.input] val_inps = node.layer.input
flag = 1
inputs = { inputs = {
"x": val_inps[0], "x": val_inps[0],
"y": val_inps[1], "y": val_inps[1],
} }
node.fluid_code.add_layer("elementwise_add", inputs=inputs, output=node) node.fluid_code.add_layer("elementwise_add", inputs=inputs, output=node)
for ipt in val_inps[2:]: for ipt in val_inps[2:]:
inputs = { inputs = {
"x": node.layer_name, "x": node.layer_name,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册