tvm
Public Member Functions | Static Public Member Functions | List of all members
tvm::relax::Function Class Reference

#include <expr.h>

Inheritance diagram for tvm::relax::Function:
Collaboration diagram for tvm::relax::Function:

Public Member Functions

 Function (Array< Var > params, Expr body, Optional< StructInfo > ret_struct_info, bool is_pure=true, DictAttrs attrs=DictAttrs(), Span span=Span())
 Construct a Relax Function. More...
 
 TVM_DEFINE_OBJECT_REF_METHODS (Function, BaseFunc, FunctionNode)
 
 TVM_DEFINE_OBJECT_REF_COW_METHOD (FunctionNode)
 
- 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)
 

Static Public Member Functions

static Function CreateEmpty (Array< Var > params, StructInfo ret_struct_info, bool is_pure=true, DictAttrs attrs=DictAttrs(), Span span=Span())
 Mimics the constructor but without body Expr. More...
 

Constructor & Destructor Documentation

◆ Function()

tvm::relax::Function::Function ( Array< Var params,
Expr  body,
Optional< StructInfo ret_struct_info,
bool  is_pure = true,
DictAttrs  attrs = DictAttrs(),
Span  span = Span() 
)
explicit

Construct a Relax Function.

Parameters
paramsThe parameters accepted by the function
bodyThe body of the function. If this is not a SeqExpr, it will be wrapped in a SeqExpr, to satisfy the Relax IR requirement that all scopes be contained in a SeqExpr.
ret_struct_infoThe StructInfo returned by the function. If std::nullopt, will be inferred from the StructInfo of the function's body.
is_pureThe purity of the function.
attrsAny attributes associated with the function. Defaults to an empty dictionary.
spanThe source span of the expression.

Member Function Documentation

◆ CreateEmpty()

static Function tvm::relax::Function::CreateEmpty ( Array< Var params,
StructInfo  ret_struct_info,
bool  is_pure = true,
DictAttrs  attrs = DictAttrs(),
Span  span = Span() 
)
static

Mimics the constructor but without body Expr.

Note
ret_struct_info is required, since it can not deduced by the body.

◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::relax::Function::TVM_DEFINE_OBJECT_REF_COW_METHOD ( FunctionNode  )

◆ TVM_DEFINE_OBJECT_REF_METHODS()

tvm::relax::Function::TVM_DEFINE_OBJECT_REF_METHODS ( Function  ,
BaseFunc  ,
FunctionNode   
)

The documentation for this class was generated from the following file: