diff --git a/python/paddle/fluid/dygraph/nn.py b/python/paddle/fluid/dygraph/nn.py index 09030e9441c89dcc95e2add3de72f4e8383bef0e..6a13e0bb49f27077f0f214895a82ba30339501a4 100644 --- a/python/paddle/fluid/dygraph/nn.py +++ b/python/paddle/fluid/dygraph/nn.py @@ -2418,6 +2418,9 @@ class PRelu(layers.Layer): default_initializer=Constant(1.0)) def forward(self, input): + if in_dygraph_mode(): + return _C_ops.prelu(input, self.weight, "NCHW", self._mode) + check_variable_and_dtype(input, 'input', ['float32'], 'PRelu') out = self._helper.create_variable_for_type_inference(self._dtype) self._helper.append_op(type="prelu", diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index cf138cc5a4f05e1904c838e634c62c817baaa6b3..d458e71fc0396ae1bc533a8c272a61ac79f67d0a 100755 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -10259,6 +10259,9 @@ def prelu(x, mode, param_attr=None, data_format="NCHW", name=None): dtype=dtype, is_bias=False, default_initializer=Constant(0.25)) + if in_dygraph_mode(): + return _C_ops.prelu(x, alpha, data_format, mode) + out = helper.create_variable_for_type_inference(dtype) helper.append_op(type="prelu", inputs={