Add extra flags for attr of affine_grid_op (#35581)

7fcb9e37 · whs · GitHub · 92810e69 · 7fcb9e37
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/operators/affine_grid_op.cc paddle/fluid/operators/affine_grid_op.cc +2 -1

未找到文件。
--- a/paddle/fluid/operators/affine_grid_op.cc
+++ b/paddle/fluid/operators/affine_grid_op.cc
@@ -138,7 +138,8 @@ class AffineGridOpMaker : public framework::OpProtoAndCheckerMaker {
    AddAttr<bool>(
        "use_cudnn",
        "(bool, default false) Only used in cudnn kernel, need install cudnn")
-        .SetDefault(true);
+        .SetDefault(true)
+        .AsExtra();
    AddAttr<bool>("align_corners",
                  "(bool, default false) Whether to align the corners of input"
                  "and ouput.")