Reduction operator. More...

#include <expr.h>

Inheritance diagram for tvm::tir::ReduceNode:

Collaboration diagram for tvm::tir::ReduceNode:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("tir.Reduce", ReduceNode, PrimExprNode)

Public Member Functions inherited from tvm::PrimExprNode
	TVM_OBJECT_ENABLE_SCRIPT_PRINTER ()

	TVM_FFI_DECLARE_OBJECT_INFO ("ir.PrimExpr", PrimExprNode, BaseExprNode)

Public Member Functions inherited from tvm::BaseExprNode
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.BaseExpr", BaseExprNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::PrimExprNode
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::BaseExprNode
static void	RegisterReflection ()

Public Attributes
CommReducer	combiner
	The commutative combiner. More...

ffi::Array< PrimExpr >	source
	The source operand. More...

ffi::Array< PrimExpr >	init
	The init operand. More...

ffi::Array< IterVar >	axis
	The reduction axis. More...

PrimExpr	condition
	Predicate on the reduction Only add the body to reduction if condition is true. More...

int	value_index
	the index of this reduce node More...

Public Attributes inherited from tvm::PrimExprNode
DataType	dtype
	The runtime data type of the primitive expression. More...

Public Attributes inherited from tvm::BaseExprNode
Span	span
	Span that points to the original source code. Reserved debug information. More...

Additional Inherited Members
Static Public Attributes inherited from tvm::PrimExprNode
static constexpr const uint32_t	_type_child_slots = 40

Static Public Attributes inherited from tvm::BaseExprNode
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

static constexpr const uint32_t	_type_child_slots = 64

Detailed Description

Reduction operator.

Member Function Documentation

◆ RegisterReflection()

static void tvm::tir::ReduceNode::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::tir::ReduceNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"tir.Reduce"	,
		ReduceNode	,
		PrimExprNode
	)

Member Data Documentation

◆ axis

ffi::Array<IterVar> tvm::tir::ReduceNode::axis

The reduction axis.

◆ combiner

CommReducer tvm::tir::ReduceNode::combiner

The commutative combiner.

◆ condition

PrimExpr tvm::tir::ReduceNode::condition

Predicate on the reduction Only add the body to reduction if condition is true.

◆ init

ffi::Array<PrimExpr> tvm::tir::ReduceNode::init

The init operand.

◆ source

ffi::Array<PrimExpr> tvm::tir::ReduceNode::source

The source operand.

◆ value_index

int tvm::tir::ReduceNode::value_index

the index of this reduce node

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

include/tvm/tir/expr.h

Public Member Functions

Static Public Member Functions

Public Attributes

Additional Inherited Members