提交 c7d1f2a8 编写于 作者: C channingss

change sum op to elementwise_add op (for paddlelite)

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