tvm::relax::Function Class Reference

#include <expr.h>

Inheritance diagram for tvm::relax::Function:

Collaboration diagram for tvm::relax::Function:

Public Member Functions
	Function (ffi::Array< Var > params, Expr body, ffi::Optional< StructInfo > ret_struct_info, bool is_pure=true, DictAttrs attrs=DictAttrs(), Span span=Span())
	Construct a Relax Function. More...
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (Function, BaseFunc, FunctionNode)
	TVM_DEFINE_OBJECT_REF_COW_METHOD (FunctionNode)
Public Member Functions inherited from tvm::BaseFunc
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (BaseFunc, RelaxExpr, BaseFuncNode)
Public Member Functions inherited from tvm::RelaxExpr
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (RelaxExpr, BaseExpr, RelaxExprNode)
Public Member Functions inherited from tvm::BaseExpr
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (BaseExpr, ObjectRef, BaseExprNode)

Static Public Member Functions
static Function	CreateEmpty (ffi::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 ( ffi::Array< Var >  params, Expr  body, ffi::Optional< StructInfo >  ret_struct_info, bool  is_pure = true, DictAttrs  attrs = DictAttrs(), Span  span = Span()  )

explicit

Construct a Relax Function.

Parameters

params	The parameters accepted by the function
body	The 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_info	The StructInfo returned by the function. If std::nullopt, will be inferred from the StructInfo of the function's body.
is_pure	The purity of the function.
attrs	Any attributes associated with the function. Defaults to an empty dictionary.
span	The source span of the expression.

Member Function Documentation

CreateEmpty()

static Function tvm::relax::Function::CreateEmpty ( ffi::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_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()

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

---

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

• include/tvm/relax/expr.h