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< Type > ret_ty, bool is_pure=true, DictAttrs attrs=DictAttrs(), Location loc=UnknownLoc())
	Construct a Relax Function.
	Function (ffi::ObjectPtr< FunctionNode > node)
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (Function, BaseFunc, FunctionNode)
	TVM_DEFINE_OBJECT_REF_COW_METHOD (FunctionNode)
Public Member Functions inherited from tvm::BaseFunc
	BaseFunc (ffi::ObjectPtr< BaseFuncNode > node)
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (BaseFunc, Expr, BaseFuncNode)
Public Member Functions inherited from tvm::Expr
bool	operator== (const Expr &other) const =delete
bool	operator!= (const Expr &other) const =delete
bool	operator< (const Expr &other) const =delete
	Expr (ffi::ObjectPtr< ExprNode > node)
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (Expr, ffi::ObjectRef, ExprNode)

Static Public Member Functions
static Function	CreateEmpty (ffi::Array< Var > params, Type ret_ty, bool is_pure=true, DictAttrs attrs=DictAttrs(), Location loc=UnknownLoc())
	Mimics the constructor but without body Expr.

Constructor & Destructor Documentation

Function() [1/2]

tvm::relax::Function::Function ( ffi::Array< Var >  params, Expr  body, ffi::Optional< Type >  ret_ty, bool  is_pure = true, DictAttrs  attrs = DictAttrs(), Location  loc = UnknownLoc()  )

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

Function() [2/2]

tvm::relax::Function::Function ( ffi::ObjectPtr< FunctionNode >  node )

inlineexplicit

Member Function Documentation

CreateEmpty()

static Function tvm::relax::Function::CreateEmpty ( ffi::Array< Var >  params, Type  ret_ty, bool  is_pure = true, DictAttrs  attrs = DictAttrs(), Location  loc = UnknownLoc()  )

static

Mimics the constructor but without body Expr.

Note

ret_ty 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_NOTNULLABLE()

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

---

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

• include/tvm/relax/expr.h