A Relax function. More...

#include <expr.h>

Inheritance diagram for tvm::relax::FunctionNode:

Collaboration diagram for tvm::relax::FunctionNode:

Public Member Functions
	TVM_DECLARE_FINAL_OBJECT_INFO (FunctionNode, BaseFuncNode)

Public Member Functions inherited from tvm::BaseFuncNode
template<typename TObjectRef >
Optional< TObjectRef >	GetAttr (const std::string &attr_key, Optional< TObjectRef > default_value=std::nullopt) const
	Get a function attribute. More...

template<typename TObjectRef >
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_DECLARE_BASE_OBJECT_INFO (BaseFuncNode, RelaxExprNode)

Public Member Functions inherited from tvm::RelaxExprNode
	TVM_DECLARE_BASE_OBJECT_INFO (RelaxExprNode, BaseExprNode)

Public Member Functions inherited from tvm::BaseExprNode
	TVM_DECLARE_BASE_OBJECT_INFO (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
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
Optional< ObjectRef >	struct_info_ = 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 constexpr const char *	_type_key = "relax.expr.Function"

Static Public Attributes inherited from tvm::BaseFuncNode
static constexpr const char *	_type_key = "ir.BaseFunc"

static constexpr const uint32_t	_type_child_slots = 2

Static Public Attributes inherited from tvm::RelaxExprNode
static constexpr const char *	_type_key = "ir.RelaxExpr"

static constexpr const uint32_t	_type_child_slots = 22

Static Public Attributes inherited from tvm::BaseExprNode
static constexpr const char *	_type_key = "ir.BaseExpr"

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

static void tvm::relax::FunctionNode::RegisterReflection ( )

inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::relax::FunctionNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	FunctionNode	,
		BaseFuncNode
	)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::relax::FunctionNode::_type_key = "relax.expr.Function"

staticconstexpr

◆ _type_s_eq_hash_kind

constexpr TVMFFISEqHashKind tvm::relax::FunctionNode::_type_s_eq_hash_kind = kTVMFFISEqHashKindDAGNode

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

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

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes