提交 96501c66 编写于 作者: W wjj19950828

fixed for ci

上级 f73d2c95
...@@ -796,11 +796,17 @@ class CaffeOpMapper(): ...@@ -796,11 +796,17 @@ class CaffeOpMapper():
inputs=inputs_dict, inputs=inputs_dict,
outputs=[node.layer_name + "_mul"]) outputs=[node.layer_name + "_mul"])
else: else:
new_shape = [1] * len(node.in_shapes[0])
new_shape[axis] = node.in_shapes[0][1]
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
"paddle.fluid.layers.elementwise_mul", "paddle.reshape",
inputs={"x": node.layer_name + "_cparam1"},
outputs=[node.layer_name + "_cparam1"],
shape=new_shape)
self.paddle_graph.add_layer(
"paddle.multiply",
inputs=inputs_dict, inputs=inputs_dict,
outputs=[node.layer_name + "_mul"], outputs=[node.layer_name + "_mul"])
axis=axis)
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
"self.create_parameter", "self.create_parameter",
inputs={}, inputs={},
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册