Managed reference to PrimFuncNode. More...

#include <function.h>

Inheritance diagram for tvm::tir::PrimFunc:

Collaboration diagram for tvm::tir::PrimFunc:

Public Member Functions
	PrimFunc (ffi::Array< tir::Var > params, Stmt body, Type ret_type=VoidType(), ffi::Map< tir::Var, Buffer > buffer_map=ffi::Map< tir::Var, Buffer >(), DictAttrs attrs=DictAttrs(), Span span=Span())
	Constructor. More...

	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (PrimFunc, BaseFunc, PrimFuncNode)

	TVM_DEFINE_OBJECT_REF_COW_METHOD (PrimFuncNode)

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)

Detailed Description

Managed reference to PrimFuncNode.

Constructor & Destructor Documentation

tvm::tir::PrimFunc::PrimFunc	(	ffi::Array< tir::Var >	params,
		Stmt	body,
		Type	ret_type = `VoidType()`,
		ffi::Map< tir::Var, Buffer >	buffer_map = `ffi::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.

tvm::tir::PrimFunc::TVM_DEFINE_OBJECT_REF_COW_METHOD ( PrimFuncNode )

tvm::tir::PrimFunc::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE	(	PrimFunc	,
		BaseFunc	,
		PrimFuncNode
	)

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