Skip to content

Commit a4d8488

Browse files
committed
to 26
Signed-off-by: Ti-Tai Wang <titaiwang@microsoft.com>
1 parent 5b72a28 commit a4d8488

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

onnx/defs/math/defs.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1891,7 +1891,7 @@ ONNX_OPERATOR_SET_SCHEMA(
18911891
defs::math::utils::MatMulShapeInference(ctx, 0, 1);
18921892
}));
18931893

1894-
static const char* CumProd_ver24_doc = R"DOC(
1894+
static const char* CumProd_ver26_doc = R"DOC(
18951895
Performs cumulative product of the input elements along the given axis.
18961896
By default, it will do the product inclusively meaning the first element is copied as is.
18971897
Through an `exclusive` attribute, this behavior can change to exclude the first element.
@@ -1915,9 +1915,9 @@ output = [6, 3, 1]
19151915

19161916
ONNX_OPERATOR_SET_SCHEMA(
19171917
CumProd,
1918-
24,
1918+
26,
19191919
OpSchema()
1920-
.SetDoc(CumProd_ver24_doc)
1920+
.SetDoc(CumProd_ver26_doc)
19211921
.Attr(
19221922
"exclusive",
19231923
"If set to 1 will return exclusive product in which the top element is not included."

onnx/defs/operator_sets.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1445,13 +1445,13 @@ class OpSet_Onnx_ver25 {
14451445
};
14461446

14471447
// Forward declarations for ai.onnx version 26
1448+
class ONNX_OPERATOR_SET_SCHEMA_CLASS_NAME(Onnx, 26, CumProd);
14481449

14491450
// Iterate over schema from ai.onnx version 26
14501451
class OpSet_Onnx_ver26 {
14511452
public:
14521453
static void ForEachSchema(std::function<void(OpSchema&&)> fn) {
1453-
// TODO: Remove after introducing the first schema to opset 26
1454-
(void)fn;
1454+
fn(GetOpSchema<ONNX_OPERATOR_SET_SCHEMA_CLASS_NAME(Onnx, 26, CumProd)>());
14551455
}
14561456
};
14571457

onnx/test/version_converter/automatic_upgrade_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -357,7 +357,7 @@ def test_Cos(self) -> None:
357357
def test_CumProd(self) -> None:
358358
self._test_op_upgrade(
359359
"CumProd",
360-
24,
360+
26,
361361
[[3, 4, 5], []],
362362
[[3, 4, 5]],
363363
[TensorProto.FLOAT, TensorProto.INT64],

0 commit comments

Comments
 (0)