I agree that layout need to be treated carefully and handling layout deserves major effort from the compilation optimization side.
Technically, the only question is how to store these information: - Types are stored in a row based format(each Expr has a type) - Layout and other attributes could be stored as a column based format (just like dataframe) in the function attribute of the function. -- You are receiving this because you are subscribed to this thread. Reply to this email directly or view it on GitHub: https://github.com/dmlc/tvm/issues/3670#issuecomment-520032503