Ah, yeah, this makes sense now.
The first LikeZapp pass will return this in certain cases: ``` tvm.relay.broadcast_to(data, list(res._checked_type_.shape)) ``` Which doesn't have a type when it is constructed, but ZeroZapp later can find that node and assume it does have a type. Thus, the problem. If you're expecting types in later passes, I think the best thing is to put InferType in your callback, or between passes as you're doing here. We could think about adding that to the rewrite infrastructure, but as I've mentioned in other threads, I don't particular want to force users to type their problems before using the pattern language in all cases. @t-vi I'll take a closer look at your examples and see if I can figure out a way to distill it into a more refined pattern. --- [Visit Topic](https://discuss.tvm.ai/t/same-shape-pattern/7012/13) 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/cd2ffedb0e4527b190ccd1b0c8f22b0abdd670b21c49cb1aaeea861ffd5aec64).