未验证 提交 8ba58eb0 编写于 作者: Z zmx 提交者: GitHub

add AsExtra to partial_sum op (#35381)

add AsExtra to the following attribute of partial_sum op:
1. use_mkldnn
上级 42d36504
...@@ -143,7 +143,8 @@ class PartialSumOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -143,7 +143,8 @@ class PartialSumOpMaker : public framework::OpProtoAndCheckerMaker {
AddAttr<bool>( AddAttr<bool>(
"use_mkldnn", "use_mkldnn",
"(bool, default false) Indicates if MKL-DNN kernel will be used") "(bool, default false) Indicates if MKL-DNN kernel will be used")
.SetDefault(false); .SetDefault(false)
.AsExtra();
AddAttr<int>("start_index", "The start index of tensor wanted to be added.") AddAttr<int>("start_index", "The start index of tensor wanted to be added.")
.SetDefault(0); .SetDefault(0);
AddAttr<int>("length", "The length of tensor wanted to be added.") AddAttr<int>("length", "The length of tensor wanted to be added.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册