diff --git a/paddle/fluid/operators/grid_sampler_op.cc b/paddle/fluid/operators/grid_sampler_op.cc index a75ea538f25561bc2d3689fd39ca3d6949d83520..0b410f07fcb576e1f68f65c2a1684e9fe0552d63 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",