ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Member Functions
luci::CircleOneHot Class Referencefinal

ONEHOT in Circle. More...

#include <CircleOneHot.h>

Collaboration diagram for luci::CircleOneHot:
Collaboration graph
[legend]

Public Member Functions

loco::Node * indices (void) const
 
void indices (loco::Node *node)
 
loco::Node * depth (void) const
 
void depth (loco::Node *node)
 
loco::Node * on_value (void) const
 
void on_value (loco::Node *node)
 
loco::Node * off_value (void) const
 
void off_value (loco::Node *node)
 
int32_t axis (void) const
 
void axis (int32_t axis)
 
- Public Member Functions inherited from luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >
 FixedArityNode ()
 
virtual ~FixedArityNode ()=default
 
uint32_t arity (void) const final
 
loco::Node * arg (uint32_t n) const final
 
void drop (void) final
 
- Public Member Functions inherited from luci::CircleNodeImpl< CircleOpcode::ONE_HOT >
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::Dialect * dialect (void) const final
 Return "Dialect" identifier that this node belongs to.
 
template<typename T >
T accept (CircleNodeVisitorBase< T > *) const
 
template<typename T >
T accept (CircleNodeMutableVisitorBase< T > *)
 
NodeName name (void) const
 
void name (const NodeName &name)
 
CircleQuantParam * quantparam (void) const
 
void quantparam (std::unique_ptr< CircleQuantParam > &&quantparam)
 
SparsityParam * sparsityparam (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 ()
 
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 luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >
loco::Use * at (uint32_t n) const
 

Detailed Description

ONEHOT in Circle.

Definition at line 31 of file CircleOneHot.h.

Member Function Documentation

◆ axis() [1/2]

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

Definition at line 48 of file CircleOneHot.h.

48{ _axis = axis; }
luci::CircleOneHot::axis
int32_t axis(void) const
Definition CircleOneHot.h:47

References axis().

Referenced by axis().

◆ axis() [2/2]

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

Definition at line 47 of file CircleOneHot.h.

47{ return _axis; }

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

◆ depth() [1/2]

void luci::CircleOneHot::depth ( loco::Node *  node)
inline

Definition at line 38 of file CircleOneHot.h.

38{ at(1)->node(node); }
loco::Use::node
Node * node(void) const
Definition Use.h:58

References luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >::at(), and loco::Use::node().

◆ depth() [2/2]

loco::Node * luci::CircleOneHot::depth ( void  ) const
inline

Definition at line 37 of file CircleOneHot.h.

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

References luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >::at(), and loco::Use::node().

◆ indices() [1/2]

void luci::CircleOneHot::indices ( loco::Node *  node)
inline

Definition at line 35 of file CircleOneHot.h.

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

References luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >::at(), and loco::Use::node().

◆ indices() [2/2]

loco::Node * luci::CircleOneHot::indices ( void  ) const
inline

Definition at line 34 of file CircleOneHot.h.

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

References luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >::at(), and loco::Use::node().

◆ off_value() [1/2]

void luci::CircleOneHot::off_value ( loco::Node *  node)
inline

Definition at line 44 of file CircleOneHot.h.

44{ at(3)->node(node); }

References luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >::at(), and loco::Use::node().

◆ off_value() [2/2]

loco::Node * luci::CircleOneHot::off_value ( void  ) const
inline

Definition at line 43 of file CircleOneHot.h.

43{ return at(3)->node(); }

References luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >::at(), and loco::Use::node().

◆ on_value() [1/2]

void luci::CircleOneHot::on_value ( loco::Node *  node)
inline

Definition at line 41 of file CircleOneHot.h.

41{ at(2)->node(node); }

References luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >::at(), and loco::Use::node().

◆ on_value() [2/2]

loco::Node * luci::CircleOneHot::on_value ( void  ) const
inline

Definition at line 40 of file CircleOneHot.h.

40{ return at(2)->node(); }

References luci::FixedArityNode< 4, CircleNodeImpl< CircleOpcode::ONE_HOT > >::at(), and loco::Use::node().

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