未验证 提交 159f10e3 编写于 作者: Z zyfncg 提交者: GitHub

clear extra atts of sequence_softmax in opmaker (#46457)

上级 4c438d30
...@@ -73,14 +73,6 @@ class SequenceSoftmaxOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -73,14 +73,6 @@ class SequenceSoftmaxOpMaker : public framework::OpProtoAndCheckerMaker {
"(bool, default false) Only used in cudnn kernel, need install cudnn") "(bool, default false) Only used in cudnn kernel, need install cudnn")
.SetDefault(false) .SetDefault(false)
.AsExtra(); .AsExtra();
AddAttr<std::string>(
"data_format",
"(string, default NCHW) Only used in "
"An optional string from: \"NHWC\", \"NCHW\". "
"Defaults to \"NHWC\". Specify the data format of the output data, "
"the input will be transformed automatically. ")
.SetDefault("AnyLayout")
.AsExtra();
AddComment(R"DOC( AddComment(R"DOC(
Sequence Softmax Operator. Sequence Softmax Operator.
......
...@@ -632,6 +632,11 @@ ...@@ -632,6 +632,11 @@
extra : extra :
attrs : [bool deterministic = false, str rng_name = "", bool force_cpu = false] attrs : [bool deterministic = false, str rng_name = "", bool force_cpu = false]
- op : sequence_softmax
backward : sequence_softmax_grad
extra :
attrs : [str data_format = "AnyLayout"]
- op : shape - op : shape
extra : extra :
attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"] attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册