Reduction operator. More...

#include <operation.h>

Inheritance diagram for tvm::te::ReduceNode:

Collaboration diagram for tvm::te::ReduceNode:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("te.Reduce", ReduceNode, OpaqueExprNode)

Public Member Functions inherited from tvm::OpaqueExprNode
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.OpaqueExpr", OpaqueExprNode, ExprNode)

Public Member Functions inherited from tvm::ExprNode
	TVM_FFI_DECLARE_OBJECT_INFO ("ir.Expr", ExprNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

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

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

Public Attributes
CommReducer	combiner
	The commutative combiner.

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

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

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

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

int	value_index
	the index of this reduce node

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

Type	ty = Type::Missing()
	The deduced or annotated type of the expression.

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

Static Public Attributes inherited from tvm::ExprNode
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::te::ReduceNode::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::te::ReduceNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"te.Reduce"	,
		ReduceNode	,
		OpaqueExprNode
	)

Member Data Documentation

◆ axis

ffi::Array<tirx::IterVar> tvm::te::ReduceNode::axis

The reduction axis.

◆ combiner

CommReducer tvm::te::ReduceNode::combiner

The commutative combiner.

◆ condition

PrimExpr tvm::te::ReduceNode::condition

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

◆ init

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

The init operand.

◆ source

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

The source operand.

◆ value_index

int tvm::te::ReduceNode::value_index

the index of this reduce node

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

include/tvm/te/operation.h

Public Member Functions

Static Public Member Functions

Public Attributes

Additional Inherited Members