IRModule: can be
TensorIR: good for multidimensional loops, tensor programs
Library Functions: manual implementation fallback
Transform:
TensorIR-level: reshape primitive function
Computational Graph level: blend library function, fusion, code generation
For more, visit: Zhihu
Table of Content