ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci::CirclePack Class Referencefinal

PACK in Circle. More...

#include <CirclePack.h>

Collaboration diagram for luci::CirclePack:

Public Member Functions

 CirclePack (uint32_t arity)
 
uint32_t values_count (void) const
 
Nodevalues (uint32_t index) const
 
void values (uint32_t index, Node *node)
 
int32_t axis (void) const
 
void axis (int32_t axis)
 
- Public Member Functions inherited from luci::VariadicArityNode< CircleNodeImpl< CircleOpcode::PACK > >
 VariadicArityNode (uint32_t arity)
 
virtual ~VariadicArityNode ()=default
 
uint32_t arity (void) const final
 
loco::Nodearg (uint32_t n) const final
 
void drop (void) final
 
- Public Member Functions inherited from luci::CircleNodeImpl< CircleOpcode::PACK >
virtual ~CircleNodeImpl ()=default
 
uint32_t opnum (void) const final
 
CircleOpcode opcode (void) const final
 
- Public Member Functions inherited from luci::CircleNode
virtual ~CircleNode ()=default
 
const loco::Dialectdialect (void) const final
 Return "Dialect" identifier that this node belongs to.
 
template<typename T >
accept (CircleNodeVisitorBase< T > *) const
 
template<typename T >
accept (CircleNodeMutableVisitorBase< T > *)
 
NodeName name (void) const
 
void name (const NodeName &name)
 
CircleQuantParamquantparam (void) const
 
void quantparam (std::unique_ptr< CircleQuantParam > &&quantparam)
 
SparsityParamsparsityparam (void) const
 
void sparsityparam (std::unique_ptr< SparsityParam > &&sparsityparam)
 
ShapeStatus shape_status (void) const
 
void shape_status (ShapeStatus ss)
 
int32_t op_version (void) const
 
void op_version (int32_t op_version)
 
- Public Member Functions inherited from loco::Node
 Node ()=default
 
 Node (const Node &)=delete
 
 Node (Node &&)=delete
 
virtual ~Node ()
 
Graphgraph (void)
 
const Graphgraph (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 luci::VariadicArityNode< CircleNodeImpl< CircleOpcode::PACK > >
loco::Useat (uint32_t n) const
 

Detailed Description

PACK in Circle.

Definition at line 33 of file CirclePack.h.

Constructor & Destructor Documentation

◆ CirclePack()

luci::CirclePack::CirclePack ( uint32_t  arity)
inline

Definition at line 36 of file CirclePack.h.

36 : VariadicArityNode<CircleNodeImpl<CircleOpcode::PACK>>(arity)
37 {
38 // TODO Support when arity is 0
39 assert(arity >= 1);
40 }
uint32_t arity(void) const final
Return the number of arguments.

References luci::VariadicArityNode< CircleNodeImpl< CircleOpcode::PACK > >::arity().

Member Function Documentation

◆ axis() [1/2]

void luci::CirclePack::axis ( int32_t  axis)
inline

Definition at line 59 of file CirclePack.h.

59{ _axis = axis; }
int32_t axis(void) const
Definition CirclePack.h:58

References axis().

Referenced by axis().

◆ axis() [2/2]

int32_t luci::CirclePack::axis ( void  ) const
inline

Definition at line 58 of file CirclePack.h.

58{ return _axis; }

Referenced by luci::CloneNodeLet< CN::OPQR >::visit(), and luci::BuiltinOptionsExtractor::visit().

◆ values() [1/2]

Node * luci::CirclePack::values ( uint32_t  index) const
inline

Definition at line 46 of file CirclePack.h.

47 {
48 assert(index < values_count());
49 return at(index)->node();
50 }
Node * node(void) const
Definition Use.h:58
uint32_t values_count(void) const
Definition CirclePack.h:43

References luci::VariadicArityNode< CircleNodeImpl< CircleOpcode::PACK > >::at(), loco::Use::node(), and values_count().

◆ values() [2/2]

void luci::CirclePack::values ( uint32_t  index,
Node node 
)
inline

Definition at line 51 of file CirclePack.h.

52 {
53 assert(index < values_count());
54 at(index)->node(node);
55 }

References luci::VariadicArityNode< CircleNodeImpl< CircleOpcode::PACK > >::at(), loco::Use::node(), and values_count().

◆ values_count()

uint32_t luci::CirclePack::values_count ( void  ) const
inline

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