A commutative reducer node to represent a commutative binary operator with identity element. More...

#include <operation.h>

Inheritance diagram for tvm::te::CommReducerNode:

Collaboration diagram for tvm::te::CommReducerNode:

Public Member Functions
ffi::Array< PrimExpr >	operator() (ffi::Array< PrimExpr > a, ffi::Array< PrimExpr > b) const
	Function call operator to combine a and b.

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("te.CommReducer", CommReducerNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
ffi::Array< PrimVar >	lhs
	The left argument of reducer.

ffi::Array< PrimVar >	rhs
	The right argument of reducer.

ffi::Array< PrimExpr >	result
	The result of reducer.

ffi::Array< PrimExpr >	identity_element
	The identity element of reducer, which leaves other elements unchanged when combined with it, with respect to the binary operation of this reducer uses.

Location	loc = UnknownLoc()
	Location that points to the original source code. Reserved debug information.

Static Public Attributes
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

Detailed Description

A commutative reducer node to represent a commutative binary operator with identity element.

Member Function Documentation

◆ operator()()

ffi::Array< PrimExpr > tvm::te::CommReducerNode::operator()	(	ffi::Array< PrimExpr >	a,
		ffi::Array< PrimExpr >	b
	)		const

Function call operator to combine a and b.

◆ RegisterReflection()

static void tvm::te::CommReducerNode::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::te::CommReducerNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"te.CommReducer"	,
		CommReducerNode	,
		ffi::Object
	)

Member Data Documentation

◆ _type_s_eq_hash_kind

constexpr TVMFFISEqHashKind tvm::te::CommReducerNode::_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

staticconstexpr

◆ identity_element

ffi::Array<PrimExpr> tvm::te::CommReducerNode::identity_element

The identity element of reducer, which leaves other elements unchanged when combined with it, with respect to the binary operation of this reducer uses.

◆ lhs

ffi::Array<PrimVar> tvm::te::CommReducerNode::lhs

The left argument of reducer.

◆ loc

Location tvm::te::CommReducerNode::loc = UnknownLoc()

mutable

Location that points to the original source code. Reserved debug information.

◆ result

ffi::Array<PrimExpr> tvm::te::CommReducerNode::result

The result of reducer.

◆ rhs

ffi::Array<PrimVar> tvm::te::CommReducerNode::rhs

The right argument of reducer.

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

Static Public Attributes