loco::TensorConcat Class Reference

Concatenate two tensors. More...

#include <Nodes.h>

Collaboration diagram for loco::TensorConcat:

Public Member Functions
Node *	lhs (void) const
void	lhs (Node *node)
Node *	rhs (void) const
void	rhs (Node *node)
uint32_t	axis (void) const
void	axis (uint32_t val)
Public Member Functions inherited from loco::CanonicalNodeDef< CanonicalOpcode::TensorConcat, FixedArity< 2 >::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

Concatenate two tensors.

Given an axis, TensorConcat takes as input two tensors and produces a tensor concatenated along the given axis.

Definition at line 531 of file Nodes.h.

Member Function Documentation

axis() [1/2]

void loco::TensorConcat::axis ( uint32_t  val )

inline

Definition at line 543 of file Nodes.h.

{ _axis = val; }

Referenced by onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::__init__(), and onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::update_state().

axis() [2/2]

uint32_t loco::TensorConcat::axis ( void  ) const

inline

Definition at line 542 of file Nodes.h.

{ return _axis; }

Referenced by onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::__init__(), exo::TensorConcatConverter::convert(), and onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::update_state().

lhs() [1/2]

void loco::TensorConcat::lhs ( Node *  node )

inline

Definition at line 536 of file Nodes.h.

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

lhs() [2/2]

Node * loco::TensorConcat::lhs ( void  ) const

inline

Definition at line 535 of file Nodes.h.

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

Referenced by exo::TensorConcatConverter::convert(), GraphTestcase< GraphCode::TensorConcat >::GraphTestcase(), and mir2loco::Transformer::visit().

rhs() [1/2]

void loco::TensorConcat::rhs ( Node *  node )

inline

Definition at line 539 of file Nodes.h.

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

rhs() [2/2]

Node * loco::TensorConcat::rhs ( void  ) const

inline

Definition at line 538 of file Nodes.h.

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

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

---

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

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