ONE - On-device Neural Engine
Loading...
Searching...
No Matches
loco::Softmax< Domain::Tensor > Class Referencefinal

Computes softmax activations for Tensor domain. More...

#include <Nodes.h>

Collaboration diagram for loco::Softmax< Domain::Tensor >:

Public Member Functions

 Softmax ()=default
 
Nodeinput (void) const
 
void input (Node *node)
 
uint32_t axis (void) const
 
void axis (uint32_t axis)
 
- Public Member Functions inherited from loco::CanonicalNodeDef< CanonicalOpcode::TensorSoftmax, FixedArity< 1 >::Mixin >
virtual ~CanonicalNodeDef ()=default
 
uint32_t opnum (void) const final
 
CanonicalOpcode opcode (void) const final
 
- Public Member Functions inherited from loco::CanonicalNode
virtual ~CanonicalNode ()=default
 
const Dialectdialect (void) const final
 Return "Dialect" identifier that this node belongs to.
 
template<typename T >
accept (CanonicalNodeVisitorBase< T > *) const
 
template<typename T >
accept (CanonicalNodeMutableVisitorBase< T > *)
 
- Public Member Functions inherited from loco::Node
 Node ()=default
 
 Node (const Node &)=delete
 
 Node (Node &&)=delete
 
virtual ~Node ()
 
Graphgraph (void)
 
const Graphgraph (void) const
 
virtual uint32_t arity (void) const =0
 Return the number of arguments.
 
virtual Nodearg (uint32_t N) const =0
 Access N-th argument node.
 
virtual void drop (void)=0
 Drop all the reference of arguments.
 
- 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.
 

Detailed Description

Computes softmax activations for Tensor domain.

Definition at line 720 of file Nodes.h.

Constructor & Destructor Documentation

◆ Softmax()

loco::Softmax< Domain::Tensor >::Softmax ( )
default

Member Function Documentation

◆ axis() [1/2]

void loco::Softmax< Domain::Tensor >::axis ( uint32_t  axis)
inline

Definition at line 731 of file Nodes.h.

731{ _axis = axis; }
uint32_t axis(void) const
Definition Nodes.h:730

References axis().

Referenced by axis().

◆ axis() [2/2]

uint32_t loco::Softmax< Domain::Tensor >::axis ( void  ) const
inline

Definition at line 730 of file Nodes.h.

730{ return _axis; }

◆ input() [1/2]

void loco::Softmax< Domain::Tensor >::input ( Node node)
inline

Definition at line 728 of file Nodes.h.

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

◆ input() [2/2]

Node * loco::Softmax< Domain::Tensor >::input ( void  ) const
inline

Definition at line 727 of file Nodes.h.

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

Referenced by mir2loco::Transformer::visit().


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