I think that's because of the function TransformShape in the file data_layout.cc. It's used to split shape like from NCHW to NCHWc. But when you want to convert your output shape from NCHW to NHWC for example, it will cause error. So you need to add your own shape transform function.
--- [Visit Topic](https://discuss.tvm.ai/t/relay-where-op-error-can-dispatch-n-nodefunctor-calls-un-registered-function-on-type-any/6075/2) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/792335054c6ca26bf6ad6319a680dd5391fe2349aef336e21eeb6ef2196c1a3f).
