From 6cdc1dc98754a466e5c8c34f8e661308a97a64ca Mon Sep 17 00:00:00 2001 From: zmx Date: Fri, 3 Sep 2021 14:07:36 +0800 Subject: [PATCH] add AsExtra to the reshape op (#35358) add AsExtra to the use_mkldnn attribute of reshape op --- paddle/fluid/operators/reshape_op.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/reshape_op.cc b/paddle/fluid/operators/reshape_op.cc index c9c1750b856..ae7e1c07b14 100644 --- a/paddle/fluid/operators/reshape_op.cc +++ b/paddle/fluid/operators/reshape_op.cc @@ -279,7 +279,8 @@ class ReshapeOpMaker : public framework::OpProtoAndCheckerMaker { .SetDefault({}); AddAttr("use_mkldnn", "(bool, default false) Only used in mkldnn kernel") - .SetDefault(false); + .SetDefault(false) + .AsExtra(); AddComment(R"DOC( Reshape Operator. -- GitLab