25 #ifndef TVM_TIR_USMP_ANALYSIS_H_ 26 #define TVM_TIR_USMP_ANALYSIS_H_ 49 #endif // TVM_TIR_USMP_ANALYSIS_H_ runtime implementation for LibTorch/TorchScript.
Definition: analyzer.h:36
BufferInfoAnalysis ExtractBufferInfo(const PrimFunc &main_func, const IRModule &mod)
Extract BufferInfo objects from a TIR IRModule.
PrimFuncFrame PrimFunc()
The primitive function statement.
Utilities for Unified Static Memory Planner.
tvm::PrimExpr mod(const tvm::PrimExpr &a, const tvm::PrimExpr &b)
Definition: broadcast.h:290
IRModuleFrame IRModule()
The IRModule declaration statement.