未验证 提交 3896bdbd 编写于 作者: S Shang Zhizhou 提交者: GitHub

add opdef extra (#35514)

* add opdef extra

* add reduce mean

* update style
上级 6f7aca9e
...@@ -679,7 +679,8 @@ class SwishOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -679,7 +679,8 @@ class SwishOpMaker : public framework::OpProtoAndCheckerMaker {
AddAttr<float>("beta", "Constant beta of swish operator").SetDefault(1.0f); AddAttr<float>("beta", "Constant beta of swish operator").SetDefault(1.0f);
AddAttr<bool>("use_mkldnn", AddAttr<bool>("use_mkldnn",
"(bool, default false) Only used in mkldnn kernel") "(bool, default false) Only used in mkldnn kernel")
.SetDefault(false); .SetDefault(false)
.AsExtra();
AddComment(R"DOC( AddComment(R"DOC(
Swish Activation Operator. Swish Activation Operator.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册