From 692ac3e5b40e6ea51cd3bc0e56f75cf4a9229cfe Mon Sep 17 00:00:00 2001 From: hong19860320 <9973393+hong19860320@users.noreply.github.com> Date: Wed, 8 Sep 2021 12:26:01 +0800 Subject: [PATCH] Add AsExtra in relu6 Op Maker (#35472) --- paddle/fluid/operators/activation_op.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/activation_op.cc b/paddle/fluid/operators/activation_op.cc index 60b3dbb25c4..69ae210148c 100644 --- a/paddle/fluid/operators/activation_op.cc +++ b/paddle/fluid/operators/activation_op.cc @@ -574,7 +574,8 @@ class Relu6OpMaker : public framework::OpProtoAndCheckerMaker { .SetDefault(6.0f); AddAttr("use_mkldnn", "(bool, default false) Only used in mkldnn kernel") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddComment(R"DOC( Relu6 Activation Operator. -- GitLab