Note that you are using `SimplifyInfernce` twice, but you want to replace the second one with `SimplifyExpr`.
But right, it seems `bias_add` and `add` are not folded. It seems `relay.transform.CanonicalizeOps()` converts `bias_add` to `add`, so you want to call it before `SimplifyExpr`. I tried your script but I didn't get satisfying output after brief attempt. Maybe you need to play around it a bit more. If you believe there is a bug / missing functionality, welcome to open a github issue. --- [Visit Topic](https://discuss.tvm.apache.org/t/batchnorm-op-fusion-in-tvm/12391/8) 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/755acbdd7f0b8091a486e0ebf6992d52f12f73840f7401f9bdb49e5a8f973f36).