I see, so are you saying the inputs in line 114 are already 5D? Or they're somehow converted to 5D?
[quote="comaniac, post:2, topic:8380"] otherwise the type won’t match [/quote] Btw, here are you saying NCHW's inputs can only be 4D and NCHWc's 5D/6D? I'm actually experimenting a customer op. How do I let it accept both 4D and 5D/6D inputs? --- [Visit Topic](https://discuss.tvm.apache.org/t/where-does-the-layout-transform-of-each-op-happen-during-alter-op-layout-pass/8380/3) 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/ba904a304b915ae690112e33450f7c43bbe064ffb2adeccd83aa977ce54002a6).