A pattern to match a Relax Function. More...
#include <dataflow_pattern.h>
Inheritance diagram for tvm::relax::FunctionPatternNode:
Collaboration diagram for tvm::relax::FunctionPatternNode:
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.dpl.FunctionPattern", FunctionPatternNode, DFPatternNode) | |
 Public Member Functions inherited from tvm::relax::DFPatternNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("relax.dpl.DFPattern", DFPatternNode, Object) | |
Static Public Member Functions | |
| static void | RegisterReflection () |
Public Attributes | |
| tvm::ffi::Array< DFPattern > | params |
| DFPattern | body |
Additional Inherited Members | |
| Static Public Attributes inherited from tvm::relax::DFPatternNode | |
| static constexpr const uint32_t | _type_child_slots = 21 |
Detailed Description
A pattern to match a Relax Function.
- See also
- Function
- FunctionPattern
Member Function Documentation
◆ RegisterReflection()
|
inlinestatic |
◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()
| tvm::relax::FunctionPatternNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "relax.dpl.FunctionPattern" | , |
| FunctionPatternNode | , | ||
| DFPatternNode | |||
| ) |
Member Data Documentation
◆ body
| DFPattern tvm::relax::FunctionPatternNode::body |
- Note
- Note that in Relax, the function body is a SeqExpr which contains 1) SeqExprNode::blocks, which is a list of blocks of statements; and 2) SeqExprNode::body, which is an Expr that can be anything. FunctionPattern only matches the body of the function (writing patterns to statements is tricky). The body of the function
◆ params
| tvm::ffi::Array<DFPattern> tvm::relax::FunctionPatternNode::params |
The parameters of the function
The documentation for this class was generated from the following file:
- include/tvm/relax/dataflow_pattern.h
