未验证 提交 7fcb9e37 编写于 作者: W whs 提交者: GitHub

Add extra flags for attr of affine_grid_op (#35581)

上级 92810e69
...@@ -138,7 +138,8 @@ class AffineGridOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -138,7 +138,8 @@ class AffineGridOpMaker : public framework::OpProtoAndCheckerMaker {
AddAttr<bool>( AddAttr<bool>(
"use_cudnn", "use_cudnn",
"(bool, default false) Only used in cudnn kernel, need install cudnn") "(bool, default false) Only used in cudnn kernel, need install cudnn")
.SetDefault(true); .SetDefault(true)
.AsExtra();
AddAttr<bool>("align_corners", AddAttr<bool>("align_corners",
"(bool, default false) Whether to align the corners of input" "(bool, default false) Whether to align the corners of input"
"and ouput.") "and ouput.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册