Ohh, I thought my bad! Thanks a lot. There is only a slight problem.
This is what I get. It adds an additional statement into relay IR ( %3 = add(%x1, %y1);) and returns the result. ``` fn (%x1: Tensor[(1, 1, 1, 20), float32], %y1: Tensor[(1, 1, 1, 20), float32]) { %0 = add(%x1, %y1); %1 = subtract(%x1, %y1); %2 = multiply(%0, %1); %3 = add(%x1, %y1); (%2, %3) } ``` This is ideally, what I'd like to create. @mbrookhart -- Is there something that I can do to fix this? ``` fn (%x1: Tensor[(1, 1, 1, 20), float32], %y1: Tensor[(1, 1, 1, 20), float32]) { %0 = add(%x1, %y1); %1 = subtract(%x1, %y1); %2 = multiply(%0, %1); (%2, %0) } ``` [quote="mbrookhart, post:6, topic:9307, full:true"] return this `super().visit_call(call)` [/quote] --- [Visit Topic](https://discuss.tvm.apache.org/t/changing-return-of-relay-function-or-irmodule/9307/7) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/fd7043df1676b5c46a00f21115858ac75c046df26a33d934cc10e8298d00c490).