ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci::FixedArityNode< N, Base > Class Template Reference

Nodes with the fixed number of inputs. More...

#include <CircleNodeMixins.h>

Collaboration diagram for luci::FixedArityNode< N, Base >:

Public Member Functions

 FixedArityNode ()
 
virtual ~FixedArityNode ()=default
 
uint32_t arity (void) const final
 
loco::Nodearg (uint32_t n) const final
 
void drop (void) final
 

Protected Member Functions

loco::Useat (uint32_t n) const
 

Detailed Description

template<uint32_t N, typename Base>
class luci::FixedArityNode< N, Base >

Nodes with the fixed number of inputs.

TODO Deprecated this class, and use loco::FixedArity instead

Definition at line 70 of file CircleNodeMixins.h.

Constructor & Destructor Documentation

◆ FixedArityNode()

template<uint32_t N, typename Base >
luci::FixedArityNode< N, Base >::FixedArityNode ( )
inline

Definition at line 73 of file CircleNodeMixins.h.

74 {
75 _args.resize(N);
76 for (uint32_t n = 0; n < N; ++n)
77 {
78 _args[n] = std::make_unique<loco::Use>(this);
79 }
80 }

◆ ~FixedArityNode()

template<uint32_t N, typename Base >
virtual luci::FixedArityNode< N, Base >::~FixedArityNode ( )
virtualdefault

Member Function Documentation

◆ arg()

template<uint32_t N, typename Base >
loco::Node * luci::FixedArityNode< N, Base >::arg ( uint32_t  n) const
inlinefinal

Definition at line 87 of file CircleNodeMixins.h.

87{ return _args.at(n)->node(); }

Referenced by luci::propagate_pad_v2_quantparam().

◆ arity()

template<uint32_t N, typename Base >
uint32_t luci::FixedArityNode< N, Base >::arity ( void  ) const
inlinefinal

Definition at line 85 of file CircleNodeMixins.h.

85{ return N; }

◆ at()

◆ drop()

template<uint32_t N, typename Base >
void luci::FixedArityNode< N, Base >::drop ( void  )
inlinefinal

Definition at line 89 of file CircleNodeMixins.h.

90 {
91 for (uint32_t n = 0; n < N; ++n)
92 {
93 _args.at(n)->node(nullptr);
94 }
95 }

Referenced by luci::fuse_sub_with_conv().


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