Hi @mbrookhart, thanks.
I got it working by replacing the last line with func = relay.Function(analysis.free_vars(new_body), new_body) Not sure, why the one that I originally used not working. But this solves the problem. Thanks. [quote="mbrookhart, post:5, topic:9254"] I think the issue is that you’re manually visiting the parameters of the function: [quote="jmatai1, post:4, topic:9254"] ``` > > ``` [/quote] [/quote] --- [Visit Topic](https://discuss.tvm.apache.org/t/rewrite-all-inputs-to-relay-ir-convert-1d-to-4d/9254/6) 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/4efacefd622cd50381443bf40efc2f1d12514bd0497035f8a8af52ba32c2c1cf).
