|
ONE - On-device Neural Engine
|
SOFTMAX in Circle. More...
#include <CircleSoftmax.h>

Public Member Functions | |
| loco::Node * | logits (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::Node * | arg (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::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) |
| CircleMXQuantParam * | mx_quantparam (void) const |
| void | mx_quantparam (std::unique_ptr< CircleMXQuantParam > &&mx_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< 1, CircleNodeImpl< CircleOpcode::SOFTMAX > > | |
| loco::Use * | at (uint32_t n) const |
SOFTMAX in Circle.
Definition at line 31 of file CircleSoftmax.h.
Definition at line 39 of file CircleSoftmax.h.
References beta().
Referenced by beta().
Definition at line 38 of file CircleSoftmax.h.
Referenced by luci::CloneNodeLet< CN::STUV >::visit(), and luci::BuiltinOptionsExtractor::visit().
|
inline |
Definition at line 35 of file CircleSoftmax.h.
References luci::FixedArityNode< 1, CircleNodeImpl< CircleOpcode::SOFTMAX > >::at(), and loco::Use::node().
|
inline |
Definition at line 34 of file CircleSoftmax.h.
References luci::FixedArityNode< 1, CircleNodeImpl< CircleOpcode::SOFTMAX > >::at(), and loco::Use::node().
Referenced by luci::sinf::Algorithm::visit().