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

GRU in Circle. More...

#include <CircleGRU.h>

Collaboration diagram for luci::CircleGRU:

Public Member Functions

loco::Nodeinput (void) const
 
void input (loco::Node *node)
 
loco::Nodehidden_hidden (void) const
 
void hidden_hidden (loco::Node *node)
 
loco::Nodehidden_hidden_bias (void) const
 
void hidden_hidden_bias (loco::Node *node)
 
loco::Nodehidden_input (void) const
 
void hidden_input (loco::Node *node)
 
loco::Nodehidden_input_bias (void) const
 
void hidden_input_bias (loco::Node *node)
 
loco::Nodestate (void) const
 
void state (loco::Node *node)
 
FusedActFunc fusedActivationFunction () const
 
void fusedActivationFunction (FusedActFunc fused_act_fun)
 
bool returnSequences () const
 
void returnSequences (bool return_sequences)
 
bool timeMajor () const
 
void timeMajor (bool time_major)
 
- Public Member Functions inherited from luci::FixedArityNode< 6, CircleNodeImpl< CircleOpcode::GRU > >
 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::GRU >
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< 6, CircleNodeImpl< CircleOpcode::GRU > >
loco::Useat (uint32_t n) const
 

Detailed Description

GRU in Circle.

Definition at line 31 of file CircleGRU.h.

Member Function Documentation

◆ fusedActivationFunction() [1/2]

FusedActFunc luci::CircleGRU::fusedActivationFunction ( ) const
inline

Definition at line 53 of file CircleGRU.h.

53{ return _fused_act_fun; }

Referenced by luci::CloneNode::visit(), and luci::BuiltinOptionsExtractor::visit().

◆ fusedActivationFunction() [2/2]

void luci::CircleGRU::fusedActivationFunction ( FusedActFunc  fused_act_fun)
inline

Definition at line 54 of file CircleGRU.h.

54{ _fused_act_fun = fused_act_fun; }

◆ hidden_hidden() [1/2]

void luci::CircleGRU::hidden_hidden ( loco::Node node)
inline

Definition at line 38 of file CircleGRU.h.

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

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

◆ hidden_hidden() [2/2]

loco::Node * luci::CircleGRU::hidden_hidden ( void  ) const
inline

Definition at line 37 of file CircleGRU.h.

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

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

◆ hidden_hidden_bias() [1/2]

void luci::CircleGRU::hidden_hidden_bias ( loco::Node node)
inline

◆ hidden_hidden_bias() [2/2]

loco::Node * luci::CircleGRU::hidden_hidden_bias ( void  ) const
inline

Definition at line 40 of file CircleGRU.h.

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

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

◆ hidden_input() [1/2]

void luci::CircleGRU::hidden_input ( loco::Node node)
inline

◆ hidden_input() [2/2]

loco::Node * luci::CircleGRU::hidden_input ( void  ) const
inline

Definition at line 43 of file CircleGRU.h.

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

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

◆ hidden_input_bias() [1/2]

void luci::CircleGRU::hidden_input_bias ( loco::Node node)
inline

◆ hidden_input_bias() [2/2]

loco::Node * luci::CircleGRU::hidden_input_bias ( void  ) const
inline

Definition at line 46 of file CircleGRU.h.

46{ return at(4)->node(); }

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

◆ input() [1/2]

void luci::CircleGRU::input ( loco::Node node)
inline

◆ input() [2/2]

loco::Node * luci::CircleGRU::input ( void  ) const
inline

Definition at line 34 of file CircleGRU.h.

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

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

◆ returnSequences() [1/2]

bool luci::CircleGRU::returnSequences ( ) const
inline

Definition at line 56 of file CircleGRU.h.

56{ return _return_sequences; }

Referenced by luci::CloneNode::visit(), and luci::BuiltinOptionsExtractor::visit().

◆ returnSequences() [2/2]

void luci::CircleGRU::returnSequences ( bool  return_sequences)
inline

Definition at line 57 of file CircleGRU.h.

57{ _return_sequences = return_sequences; }

◆ state() [1/2]

void luci::CircleGRU::state ( loco::Node node)
inline

◆ state() [2/2]

loco::Node * luci::CircleGRU::state ( void  ) const
inline

Definition at line 49 of file CircleGRU.h.

49{ return at(5)->node(); }

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

◆ timeMajor() [1/2]

bool luci::CircleGRU::timeMajor ( ) const
inline

Definition at line 59 of file CircleGRU.h.

59{ return _time_major; }

Referenced by luci::CloneNode::visit(), and luci::BuiltinOptionsExtractor::visit().

◆ timeMajor() [2/2]

void luci::CircleGRU::timeMajor ( bool  time_major)
inline

Definition at line 60 of file CircleGRU.h.

60{ _time_major = time_major; }

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