tvm::tir::CommReducerNode Class Reference

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

#include <expr.h>

Inheritance diagram for tvm::tir::CommReducerNode:

Collaboration diagram for tvm::tir::CommReducerNode:

Public Member Functions
Array< PrimExpr >	operator() (Array< PrimExpr > a, Array< PrimExpr > b) const
	Function call operator to combine a and b. More...
	TVM_DECLARE_FINAL_OBJECT_INFO (CommReducerNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
Array< Var >	lhs
	The left argument of reducer. More...
Array< Var >	rhs
	The right argument of reducer. More...
Array< PrimExpr >	result
	The result of reducer. More...
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. More...
Span	span
	Span that points to the original source code. Reserved debug information. More...

Static Public Attributes
static constexpr const char *	_type_key = "tir.CommReducer"
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()()

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

Function call operator to combine a and b.

RegisterReflection()

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

inlinestatic

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::tir::CommReducerNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	CommReducerNode	,
		Object
	)

Member Data Documentation

_type_key

constexpr const char* tvm::tir::CommReducerNode::_type_key = "tir.CommReducer"

staticconstexpr

_type_s_eq_hash_kind

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

staticconstexpr

identity_element

Array<PrimExpr> tvm::tir::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

Array<Var> tvm::tir::CommReducerNode::lhs

The left argument of reducer.

result

Array<PrimExpr> tvm::tir::CommReducerNode::result

The result of reducer.

rhs

Array<Var> tvm::tir::CommReducerNode::rhs

The right argument of reducer.

span

Span tvm::tir::CommReducerNode::span

mutable

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

---

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

• include/tvm/tir/expr.h