Try this:
```
new_body = self.visit(fn.body)
print("Visited all", new_body)
return_values_to_function = relay.Tuple([new_body] + self.return_values)
func = relay.Function(fn.params, return_values_to_function,
fn.ret_type, fn.type_params, fn.attrs)
```
---
[Visit
Topic](https://discuss.tvm.apache.org/t/changing-return-of-relay-function-or-irmodule/9307/4)
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/b2e20e197f3907a66bfe73cf49e3efc6764003446f88d001101e3d79a7380efe).
