loco::TensorReduce Class Reference

Computes ReduceFunc operations for Tensor domain. More...

#include <Nodes.h>

Collaboration diagram for loco::TensorReduce:

Public Member Functions
Node *	input (void) const
void	input (Node *node)
const TensorAxisSet *	axes (void) const
TensorAxisSet *	axes (void)
ReduceFunc	func (void) const
void	func (ReduceFunc func)
Public Member Functions inherited from loco::CanonicalNodeDef< CanonicalOpcode::TensorReduce, FixedArity< 1 >::Mixin >
virtual	~CanonicalNodeDef ()=default
uint32_t	opnum (void) const final
CanonicalOpcode	opcode (void) const final
Public Member Functions inherited from loco::CanonicalNode
virtual	~CanonicalNode ()=default
const Dialect *	dialect (void) const final
	Return "Dialect" identifier that this node belongs to.
template<typename T >
T	accept (CanonicalNodeVisitorBase< T > *) const
template<typename T >
T	accept (CanonicalNodeMutableVisitorBase< T > *)
Public Member Functions inherited from loco::Node
	Node ()=default
	Node (const Node &)=delete
	Node (Node &&)=delete
virtual	~Node ()
Graph *	graph (void)
const Graph *	graph (void) const
virtual uint32_t	arity (void) const =0
	Return the number of arguments.
virtual Node *	arg (uint32_t N) const =0
	Access N-th argument node.
virtual void	drop (void)=0
	Drop all the reference of arguments.
Public Member Functions inherited from loco::AnnotatedItem< NodeAnnotation >
	AnnotatedItem ()=default
virtual	~AnnotatedItem ()=default
const T *	annot (void) const
	Retrieve a stored annotation of type T.
void	annot (std::unique_ptr< T > &&p)
	Attach or remove a new annotation of type T.

Detailed Description

Computes ReduceFunc operations for Tensor domain.

Note

All the reduce functions always keep dimensions

Definition at line 618 of file Nodes.h.

Member Function Documentation

axes() [1/2]

TensorAxisSet * loco::TensorReduce::axes ( void  )

inline

Definition at line 627 of file Nodes.h.

{ return &_axes; }

axes() [2/2]

const TensorAxisSet * loco::TensorReduce::axes ( void  ) const

inline

Definition at line 626 of file Nodes.h.

{ return &_axes; }

func() [1/2]

void loco::TensorReduce::func ( ReduceFunc  func )

inline

Definition at line 631 of file Nodes.h.

{ _func = func; }

References func().

Referenced by func().

func() [2/2]

ReduceFunc loco::TensorReduce::func ( void  ) const

inline

Definition at line 630 of file Nodes.h.

{ return _func; }

Referenced by exo::TensorReduceConverter::convert().

input() [1/2]

void loco::TensorReduce::input ( Node *  node )

inline

Definition at line 623 of file Nodes.h.

{ at(0)->node(node); }

input() [2/2]

Node * loco::TensorReduce::input ( void  ) const

inline

Definition at line 622 of file Nodes.h.

{ return at(0)->node(); }

---

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

• compiler/loco/include/loco/IR/Nodes.h