Thanks, yeah I also remember `te.scan` only supports scanning along the first axis. I think `te.extern` is good as a first step.
--- [Visit Topic](https://discuss.tvm.apache.org/t/add-some-new-tensorflow-ops/8217/7) 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/89b433e65aed65c502ede08563f84813cc0b0386a3b7efb60dd3077d5b4b76f9).