Skip to content

Commit

Permalink
modify attr
Browse files Browse the repository at this point in the history
  • Loading branch information
ajindal1 committed May 21, 2024
1 parent c028b32 commit 0d53007
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions onnxruntime/core/graph/contrib_ops/contrib_defs.cc
Original file line number Diff line number Diff line change
Expand Up @@ -639,8 +639,8 @@ ONNX_MS_OPERATOR_SET_SCHEMA(
.SetDomain(kMSDomain)
.SinceVersion(1)
.SetDoc(S2SModelSplitQuickGelu_ver1_doc)
.Attr("axis", "axis value for split", AttributeProto::INT, static_cast<int64_t>(-1))
.Attr("alpha", "Alpha value.", AttributeProto::FLOAT, 1.702f)
// .Attr("axis", "axis value for split", AttributeProto::INT, static_cast<int64_t>(-1))
// .Attr("alpha", "Alpha value.", AttributeProto::FLOAT, 1.702f)
.Input(0, "X", "The input data as Tensor.", "T")
// .Input(1, "split", "split tensor.", "T")
.Output(0, "Y", "The output.", "T")
Expand Down

0 comments on commit 0d53007

Please sign in to comment.