From df57df94f83410d7a97b8eda508b4fa7aa5e5f0e Mon Sep 17 00:00:00 2001 From: LielinJiang <50691816+LielinJiang@users.noreply.github.com> Date: Wed, 1 Sep 2021 22:23:00 +0800 Subject: [PATCH] add AsExtra for grid_sampler_op (#35339) --- paddle/fluid/operators/grid_sampler_op.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/grid_sampler_op.cc b/paddle/fluid/operators/grid_sampler_op.cc index a75ea538f25..0b410f07fcb 100644 --- a/paddle/fluid/operators/grid_sampler_op.cc +++ b/paddle/fluid/operators/grid_sampler_op.cc @@ -100,7 +100,8 @@ class GridSampleOpMaker : public framework::OpProtoAndCheckerMaker { AddAttr( "use_cudnn", "(bool, default true) Only used in cudnn kernel, need install cudnn") - .SetDefault(true); + .SetDefault(true) + .AsExtra(); AddAttr( "align_corners", -- GitLab