提交 8069fff5 编写于 作者: W wjj19950828

rm fluid for channel_shuffle

上级 ff6c6996
...@@ -1209,10 +1209,10 @@ class CaffeOpMapper(): ...@@ -1209,10 +1209,10 @@ class CaffeOpMapper():
input = self.graph.get_input_node(node, idx=0, copy=True) input = self.graph.get_input_node(node, idx=0, copy=True)
params = node.layer.shuffle_channel_param params = node.layer.shuffle_channel_param
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
"paddle.fluid.layers.shuffle_channel", "paddle.nn.functional.channel_shuffle",
inputs={"x": input.name}, inputs={"x": input.name},
outputs=[node.layer_name], outputs=[node.layer_name],
group=params.group) groups=params.group)
def Upsample(self, node): def Upsample(self, node):
assert len( assert len(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册