@@ -37,7 +37,7 @@ def light_api(
3737 op: "GraphBuilder",
3838 X: "FLOAT[]",
3939 ):
40- Y = op.Exp(X)
40+ Y = op.Exp(X, outputs=['Y'] )
4141 op.Identity(Y, outputs=["Y"])
4242 return Y
4343
@@ -57,7 +57,7 @@ def light_api(
5757op :"GraphBuilder" ,
5858X :"FLOAT[]" ,# noqa: F722
5959 ):
60- Y = op .Exp (X )
60+ Y = op .Exp (X , outputs = [ "Y" ] )
6161op .Identity (Y ,outputs = ["Y" ])
6262return Y
6363
@@ -93,8 +93,8 @@ def light_api(
9393 X: "FLOAT[]",
9494 ):
9595 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'] )
9898 op.Identity(Y, outputs=["Y"])
9999 return Y
100100
@@ -148,7 +148,7 @@ def light_api(
148148 op: "GraphBuilder",
149149 X: "FLOAT[]",
150150 ):
151- Y = op.Exp(X)
151+ Y = op.Exp(X, outputs=['Y'] )
152152 op.Identity(Y, outputs=["Y"])
153153 return Y
154154
@@ -240,8 +240,8 @@ def example(
240240 A: "FLOAT[, ]",
241241 B: "FLOAT[, ]",
242242 ):
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'] )
245245 op.Identity(Y, outputs=["Y"])
246246 return Y
247247
@@ -252,8 +252,8 @@ def make_custom_LinearRegression(g: "GraphBuilder"):
252252 a = gr.make_tensor_input('a')
253253 b = gr.make_tensor_input('b')
254254 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'] )
257257 gr.make_tensor_output(y)
258258 g.add_function(builder=gr)
259259 return gr