未验证 提交 df57df94 编写于 作者: L LielinJiang 提交者: GitHub

add AsExtra for grid_sampler_op (#35339)

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