|
tvm
|
PassNode is the base type of differnt types of optimization passes. It is designed as a pure class and implemented by different pass subclasses at different granularity of Relax nodes. More...
#include <transform.h>
Public Member Functions | |
| virtual | ~PassNode () |
| virtual PassInfo | Info () const =0 |
| Get the pass information/meta data. | |
| IRModule | operator() (IRModule mod) const |
| Transform mod using the default PassContext in the current scope. | |
| virtual IRModule | operator() (IRModule mod, const PassContext &pass_ctx) const =0 |
| Transform mod using a functor under a given pass context. | |
| TVM_FFI_DECLARE_OBJECT_INFO ("transform.Pass", PassNode, ffi::Object) | |
PassNode is the base type of differnt types of optimization passes. It is designed as a pure class and implemented by different pass subclasses at different granularity of Relax nodes.
|
inlinevirtual |
Get the pass information/meta data.
Implemented in tvm::transform::SequentialNode.
Transform mod using the default PassContext in the current scope.
| mod | The module that an optimization pass runs on. |
|
pure virtual |
Transform mod using a functor under a given pass context.
| mod | The module that an optimization pass runs on. |
| pass_ctx | The pass context that can provide information for the optimization. |
Implemented in tvm::transform::SequentialNode.
| tvm::transform::PassNode::TVM_FFI_DECLARE_OBJECT_INFO | ( | "transform.Pass" | , |
| PassNode | , | ||
| ffi::Object | |||
| ) |