@@ -37,7 +37,7 @@ def light_api(
37
37
op: "GraphBuilder",
38
38
X: "FLOAT[]",
39
39
):
40
- Y = op.Exp(X)
40
+ Y = op.Exp(X, outputs=['Y'] )
41
41
op.Identity(Y, outputs=["Y"])
42
42
return Y
43
43
@@ -57,7 +57,7 @@ def light_api(
57
57
op : "GraphBuilder" ,
58
58
X : "FLOAT[]" , # noqa: F722
59
59
):
60
- Y = op .Exp (X )
60
+ Y = op .Exp (X , outputs = [ "Y" ] )
61
61
op .Identity (Y , outputs = ["Y" ])
62
62
return Y
63
63
@@ -93,8 +93,8 @@ def light_api(
93
93
X: "FLOAT[]",
94
94
):
95
95
r = np.array([-1, 1], dtype=np.int64)
96
- r0_0 = op.Reshape(X, r)
97
- Y = op.Transpose(r0_0, perm=[1, 0])
96
+ r0_0 = op.Reshape(X, r, outputs=['r0_0'] )
97
+ Y = op.Transpose(r0_0, perm=[1, 0], outputs=['Y'] )
98
98
op.Identity(Y, outputs=["Y"])
99
99
return Y
100
100
@@ -148,7 +148,7 @@ def light_api(
148
148
op: "GraphBuilder",
149
149
X: "FLOAT[]",
150
150
):
151
- Y = op.Exp(X)
151
+ Y = op.Exp(X, outputs=['Y'] )
152
152
op.Identity(Y, outputs=["Y"])
153
153
return Y
154
154
@@ -240,8 +240,8 @@ def example(
240
240
A: "FLOAT[, ]",
241
241
B: "FLOAT[, ]",
242
242
):
243
- Y1 = op.LinearRegression(X, A, B, domain='custom')
244
- Y = op.Abs(Y1)
243
+ Y1 = op.LinearRegression(X, A, B, domain='custom', outputs=['Y1'] )
244
+ Y = op.Abs(Y1, outputs=['Y'] )
245
245
op.Identity(Y, outputs=["Y"])
246
246
return Y
247
247
@@ -252,8 +252,8 @@ def make_custom_LinearRegression(g: "GraphBuilder"):
252
252
a = gr.make_tensor_input('a')
253
253
b = gr.make_tensor_input('b')
254
254
op = gr.op
255
- xa = op.MatMul(x, a)
256
- y = op.Add(xa, b)
255
+ xa = op.MatMul(x, a, outputs=['xa'] )
256
+ y = op.Add(xa, b, outputs=['y'] )
257
257
gr.make_tensor_output(y)
258
258
g.add_function(builder=gr)
259
259
return gr
0 commit comments