The approach I suggested is the most straightforward one. Relay to TIR is not one-to-one mapping. A Relay node may be lowered to different TIR functions for different target and input shapes/dtype.
--- [Visit Topic](https://discuss.tvm.apache.org/t/profile-on-relay-level/9568/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/11f0dc68bb90cfb598b7f1e413cb36d2860c5b842d12800e4e82f2a8bfa2be4c).