Managed reference to PrimFuncNode. More...
#include <function.h>
Inheritance diagram for tvm::tir::PrimFunc:
Collaboration diagram for tvm::tir::PrimFunc:
Public Member Functions | |
| PrimFunc (Array< tir::Var > params, Stmt body, Type ret_type=VoidType(), Map< tir::Var, Buffer > buffer_map=Map< tir::Var, Buffer >(), DictAttrs attrs=DictAttrs(), Span span=Span()) | |
| Constructor. More... | |
| TVM_DEFINE_OBJECT_REF_METHODS (PrimFunc, BaseFunc, PrimFuncNode) | |
| TVM_DEFINE_OBJECT_REF_COW_METHOD (PrimFuncNode) | |
 Public Member Functions inherited from tvm::BaseFunc | |
| TVM_DEFINE_OBJECT_REF_METHODS (BaseFunc, RelaxExpr, BaseFuncNode) | |
| Public Member Functions inherited from tvm::RelaxExpr | |
| TVM_DEFINE_OBJECT_REF_METHODS (RelaxExpr, BaseExpr, RelaxExprNode) | |
| Public Member Functions inherited from tvm::BaseExpr | |
| TVM_DEFINE_OBJECT_REF_METHODS (BaseExpr, ObjectRef, BaseExprNode) | |
Detailed Description
Managed reference to PrimFuncNode.
- See also
- PrimFuncNode
Constructor & Destructor Documentation
◆ PrimFunc()
| tvm::tir::PrimFunc::PrimFunc | ( | Array< tir::Var > | params, |
| Stmt | body, | ||
| Type | ret_type = VoidType(), |
||
| Map< tir::Var, Buffer > | buffer_map = Map< tir::Var, Buffer >(), |
||
| DictAttrs | attrs = DictAttrs(), |
||
| Span | span = Span() |
||
| ) |
Constructor.
- Parameters
-
params The parameters of the function. body The body of the function. ret_type The return type of the function. buffer_map The buffer map for parameter buffer unpacking. This contains buffer objects as they appear in the body of the PrimFunc. (e.g. a buffer of shape [1024]originally generated as a tensor of shape[32, 32])attrs Additional function attributes. span The location of this object in the source code.
Member Function Documentation
◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()
| tvm::tir::PrimFunc::TVM_DEFINE_OBJECT_REF_COW_METHOD | ( | PrimFuncNode | ) |
◆ TVM_DEFINE_OBJECT_REF_METHODS()
| tvm::tir::PrimFunc::TVM_DEFINE_OBJECT_REF_METHODS | ( | PrimFunc | , |
| BaseFunc | , | ||
| PrimFuncNode | |||
| ) |
The documentation for this class was generated from the following file:
- include/tvm/tir/function.h
