tvm::tir::ReduceNode Class Reference

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