1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
from onnx import TensorProto
from onnx.helper import make_tensor
from onnxscript import opset15 as op
from onnxscript import script
@script()
def sumprod(x, N):
sum = op.Identity(x)
prod = op.Identity(x)
cond = op.Constant(value=make_tensor("true", TensorProto.BOOL, [1], [1]))
i = op.Constant(value=make_tensor("i", TensorProto.INT64, [1], [0]))
while cond:
sum = sum + x
prod = prod * x
i = i + 1
cond = i < 10
return sum, prod
|
