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

SOFTMAX in Circle. More...

#include <CircleSoftmax.h>

Collaboration diagram for luci::CircleSoftmax:

Public Member Functions

loco::Nodelogits (void) const
 
void logits (loco::Node *node)
 
float beta (void) const
 
void beta (float beta)
 
- Public Member Functions inherited from luci::FixedArityNode< 1, CircleNodeImpl< CircleOpcode::SOFTMAX > >
 FixedArityNode ()
 
virtual ~FixedArityNode ()=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::SOFTMAX >
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::FixedArityNode< 1, CircleNodeImpl< CircleOpcode::SOFTMAX > >
loco::Useat (uint32_t n) const
 

Detailed Description

SOFTMAX in Circle.

Definition at line 31 of file CircleSoftmax.h.

Member Function Documentation

◆ beta() [1/2]

void luci::CircleSoftmax::beta ( float  beta)
inline

Definition at line 39 of file CircleSoftmax.h.

39{ _beta = beta; }
float beta(void) const

References beta().

Referenced by beta().

◆ beta() [2/2]

float luci::CircleSoftmax::beta ( void  ) const
inline

Definition at line 38 of file CircleSoftmax.h.

38{ return _beta; }

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

◆ logits() [1/2]

void luci::CircleSoftmax::logits ( loco::Node node)
inline

◆ logits() [2/2]

loco::Node * luci::CircleSoftmax::logits ( void  ) const
inline

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