It probably require working on the tensor ir, as the tvm compute level is pure.
-- You are receiving this because you are subscribed to this thread. Reply to this email directly or view it on GitHub: https://github.com/apache/incubator-tvm/issues/1996#issuecomment-596050218