未验证 提交 aea9546b 编写于 作者: S SunAhong1993 提交者: GitHub

Update adaptive_pool2d_fuser.py

上级 f64b23a7
...@@ -167,7 +167,7 @@ class DygraphAdaptivePool2dFuser(FuseBase): ...@@ -167,7 +167,7 @@ class DygraphAdaptivePool2dFuser(FuseBase):
new_layer = PaddleLayer( new_layer = PaddleLayer(
layers_id[0], layers_id[0],
"paddle.nn.functional.adaptive_avg_pool2d", "paddle.nn.functional.adaptive_avg_pool2d",
inputs={"input": input_name}, inputs={"x": input_name},
outputs=[output_name], outputs=[output_name],
**attrs) **attrs)
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册