moco::TFPack Class Reference

#include <TFPack.h>

Collaboration diagram for moco::TFPack:

Public Member Functions
	TFPack (uint32_t arity)
Node *	values (uint32_t index) const
void	values (uint32_t index, Node *node)
uint32_t	N (void) const
int32_t	axis (void) const
void	axis (int32_t axis)
Public Member Functions inherited from moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >
	VariadicArityNode (uint32_t arity)
virtual	~VariadicArityNode ()=default
uint32_t	arity (void) const final
loco::Node *	arg (uint32_t n) const final
void	drop (void) final
Public Member Functions inherited from moco::TFNodeImpl< TFOpcode::Pack >
virtual	~TFNodeImpl ()=default
uint32_t	opnum (void) const final
TFOpcode	opcode (void) const final
Public Member Functions inherited from moco::TFNode
virtual	~TFNode ()=default
const loco::Dialect *	dialect (void) const final
	Return "Dialect" identifier that this node belongs to.
template<typename T >
T	accept (TFNodeVisitorBase< T > *) const
template<typename T >
T	accept (TFNodeMutableVisitorBase< T > *)
NodeName	name (void) const
void	name (const NodeName &name)
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
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.

Additional Inherited Members
Protected Member Functions inherited from moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >
loco::Use *	at (uint32_t n) const

Detailed Description

Note

TFPack corresponds to the following GraphDef

Definition at line 53 of file TFPack.h.

Constructor & Destructor Documentation

TFPack()

moco::TFPack::TFPack ( uint32_t  arity )

inline

Definition at line 56 of file TFPack.h.

                         : VariadicArityNode<TFNodeImpl<TFOpcode::Pack>>(arity)
  {
    // at least one item should exist
    assert(arity >= 1);
  }

References moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::arity().

Member Function Documentation

axis() [1/2]

void moco::TFPack::axis ( int32_t  axis )

inline

Definition at line 78 of file TFPack.h.

{ _axis = axis; }

References axis().

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

axis() [2/2]

int32_t moco::TFPack::axis ( void  ) const

inline

Definition at line 77 of file TFPack.h.

{ return _axis; }

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

N()

uint32_t moco::TFPack::N ( void  ) const

inline

Definition at line 75 of file TFPack.h.

{ return arity(); }

References moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::arity().

values() [1/2]

Node * moco::TFPack::values ( uint32_t  index ) const

inline

Definition at line 63 of file TFPack.h.

  {
    assert(index < arity());
    return at(index)->node();
  }

References moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::arity(), moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::at(), moco::index(), and loco::Use::node().

values() [2/2]

void moco::TFPack::values ( uint32_t  index, Node *  node  )

inline

Definition at line 68 of file TFPack.h.

  {
    assert(index < arity());
    at(index)->node(node);
  }

References moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::arity(), moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::at(), moco::index(), and loco::Use::node().

---

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

• compiler/moco/lang/include/moco/IR/Nodes/TFPack.h