1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
from onnx import TensorProto
from onnx.helper import make_tensor
from onnxscript import script
from onnxscript.onnx_opset import opset15 as op
from onnxscript.onnx_types import FLOAT
# same variable assigned multiple times
@script()
def loop1(A: FLOAT["N"]) -> FLOAT["N"]:
T = A
for i in range(10):
T = T + A * op.Cast(i, to=TensorProto.FLOAT)
return T
@script()
def loop_range_cond_only(A: FLOAT["N"]) -> FLOAT["N"]:
T = A
cond = op.Constant(value=make_tensor("true", TensorProto.BOOL, [], [1]))
while cond:
T = T + A
cond = op.ReduceSum(T) > -10
return T
|
