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_DECLARE_FINAL_OBJECT_INFO (ReduceNode, PrimExprNode)
Public Member Functions inherited from tvm::PrimExprNode
	TVM_OBJECT_ENABLE_SCRIPT_PRINTER ()
	TVM_DECLARE_BASE_OBJECT_INFO (PrimExprNode, 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::PrimExprNode
static void	RegisterReflection ()
Static Public Member Functions inherited from tvm::BaseExprNode
static void	RegisterReflection ()

Public Attributes
CommReducer	combiner
	The commutative combiner. More...
Array< PrimExpr >	source
	The source operand. More...
Array< PrimExpr >	init
	The init operand. More...
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...

Static Public Attributes
static constexpr const char *	_type_key = "tir.Reduce"
Static Public Attributes inherited from tvm::PrimExprNode
static constexpr const char *	_type_key = "ir.PrimExpr"
static constexpr const uint32_t	_type_child_slots = 40
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

Reduction operator.

Member Function Documentation

RegisterReflection()

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

inlinestatic

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::tir::ReduceNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	ReduceNode	,
		PrimExprNode
	)

Member Data Documentation

_type_key

constexpr const char* tvm::tir::ReduceNode::_type_key = "tir.Reduce"

staticconstexpr

axis

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

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

The init operand.

source

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