A Relax function. More...
#include <expr.h>
Inheritance diagram for tvm::relax::FunctionNode:
Collaboration diagram for tvm::relax::FunctionNode:
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.expr.Function", FunctionNode, BaseFuncNode) | |
 Public Member Functions inherited from tvm::BaseFuncNode | |
| template<typename TObjectRef > | |
| ffi::Optional< TObjectRef > | GetAttr (const std::string &attr_key, ffi::Optional< TObjectRef > default_value=std::nullopt) const |
| Get a function attribute. More... | |
| template<typename TObjectRef > | |
| ffi::Optional< TObjectRef > | GetAttr (const std::string &attr_key, TObjectRef default_value) const |
| bool | HasNonzeroAttr (const std::string &attr_key) const |
| Check whether the function has an non-zero integer attr. More... | |
| LinkageType | GetLinkageType () const |
| Get the type of the linkage. More... | |
| TVM_FFI_DECLARE_OBJECT_INFO ("ir.BaseFunc", BaseFuncNode, RelaxExprNode) | |
| Public Member Functions inherited from tvm::RelaxExprNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("ir.RelaxExpr", RelaxExprNode, BaseExprNode) | |
| Public Member Functions inherited from tvm::BaseExprNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("ir.BaseExpr", BaseExprNode, Object) | |
Static Public Member Functions | |
| static void | RegisterReflection () |
| Static Public Member Functions inherited from tvm::BaseFuncNode | |
| static void | RegisterReflection () |
| Static Public Member Functions inherited from tvm::RelaxExprNode | |
| static void | RegisterReflection () |
| Static Public Member Functions inherited from tvm::BaseExprNode | |
| static void | RegisterReflection () |
Public Attributes | |
| ffi::Array< Var > | params |
| The parameters to the function. More... | |
| SeqExpr | body |
| The body of the function. More... | |
| StructInfo | ret_struct_info |
| The return type of the function. More... | |
| bool | is_pure |
| Whether the function is annotated as pure or not. More... | |
| Public Attributes inherited from tvm::BaseFuncNode | |
| DictAttrs | attrs |
| Additional attributes storing the meta-data. More... | |
| Public Attributes inherited from tvm::RelaxExprNode | |
| ffi::Optional< ObjectRef > | struct_info_ = ffi::Optional<ObjectRef>() |
| Stores the result of structure information of the expression that encapsulate both static shape and runtime information such as shape. More... | |
| Public Attributes inherited from tvm::BaseExprNode | |
| Span | span |
| Span that points to the original source code. Reserved debug information. More... | |
Static Public Attributes | |
| static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindDAGNode |
| Static Public Attributes inherited from tvm::BaseFuncNode | |
| static constexpr const uint32_t | _type_child_slots = 2 |
| Static Public Attributes inherited from tvm::RelaxExprNode | |
| static constexpr const uint32_t | _type_child_slots = 22 |
| Static Public Attributes inherited from tvm::BaseExprNode | |
| static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode |
| static constexpr const uint32_t | _type_child_slots = 64 |
Detailed Description
A Relax function.
Member Function Documentation
◆ RegisterReflection()
|
inlinestatic |
◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()
| tvm::relax::FunctionNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "relax.expr.Function" | , |
| FunctionNode | , | ||
| BaseFuncNode | |||
| ) |
Member Data Documentation
◆ _type_s_eq_hash_kind
|
staticconstexpr |
◆ body
| SeqExpr tvm::relax::FunctionNode::body |
The body of the function.
◆ is_pure
| bool tvm::relax::FunctionNode::is_pure |
Whether the function is annotated as pure or not.
◆ params
| ffi::Array<Var> tvm::relax::FunctionNode::params |
The parameters to the function.
◆ ret_struct_info
| StructInfo tvm::relax::FunctionNode::ret_struct_info |
The return type of the function.
The documentation for this class was generated from the following file:
- include/tvm/relax/expr.h
