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

#include <TFFusedBatchNorm.h>

Collaboration diagram for moco::TFFusedBatchNorm:
Collaboration graph
[legend]

Public Member Functions

 TFFusedBatchNorm ()=default
 
Node * x (void) const
 
void x (Node *node)
 
Node * scale (void) const
 
void scale (Node *node)
 
Node * offset (void) const
 
void offset (Node *node)
 
Node * mean (void) const
 
void mean (Node *node)
 
Node * variance (void) const
 
void variance (Node *node)
 
float epsilon (void) const
 
void epsilon (float epsilon)
 
- Public Member Functions inherited from moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >
 FixedArityNode ()
 
virtual ~FixedArityNode ()=default
 
unsigned arity (void) const final
 
loco::Node * arg (uint32_t n) const final
 
void drop (void) final
 
- Public Member Functions inherited from moco::TFNodeImpl< TFOpcode::FusedBatchNorm >
virtual ~TFNodeImpl ()=default
 
uint32_t opnum (void) const final
 
TFOpcode opcode (void) const final
 
- Public Member Functions inherited from moco::TFNode
virtual ~TFNode ()=default
 
const loco::Dialect * dialect (void) const final
 Return "Dialect" identifier that this node belongs to.
 
template<typename T >
T accept (TFNodeVisitorBase< T > *) const
 
template<typename T >
T accept (TFNodeMutableVisitorBase< T > *)
 
NodeName name (void) const
 
void name (const NodeName &name)
 
- 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 moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >
loco::Use * at (unsigned n) const
 

Detailed Description

Definition at line 25 of file TFFusedBatchNorm.h.

Constructor & Destructor Documentation

◆ TFFusedBatchNorm()

moco::TFFusedBatchNorm::TFFusedBatchNorm ( )
default

Member Function Documentation

◆ epsilon() [1/2]

void moco::TFFusedBatchNorm::epsilon ( float  epsilon)
inline

Definition at line 47 of file TFFusedBatchNorm.h.

47{ _epsilon = epsilon; }
moco::TFFusedBatchNorm::epsilon
float epsilon(void) const
Definition TFFusedBatchNorm.h:46

References epsilon().

Referenced by epsilon().

◆ epsilon() [2/2]

float moco::TFFusedBatchNorm::epsilon ( void  ) const
inline

Definition at line 46 of file TFFusedBatchNorm.h.

46{ return _epsilon; }

◆ mean() [1/2]

void moco::TFFusedBatchNorm::mean ( Node *  node)
inline

Definition at line 41 of file TFFusedBatchNorm.h.

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

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

◆ mean() [2/2]

Node * moco::TFFusedBatchNorm::mean ( void  ) const
inline

Definition at line 40 of file TFFusedBatchNorm.h.

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

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

◆ offset() [1/2]

void moco::TFFusedBatchNorm::offset ( Node *  node)
inline

Definition at line 38 of file TFFusedBatchNorm.h.

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

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

◆ offset() [2/2]

Node * moco::TFFusedBatchNorm::offset ( void  ) const
inline

Definition at line 37 of file TFFusedBatchNorm.h.

37{ return at(2)->node(); } // beta

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

◆ scale() [1/2]

void moco::TFFusedBatchNorm::scale ( Node *  node)
inline

Definition at line 35 of file TFFusedBatchNorm.h.

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

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

◆ scale() [2/2]

Node * moco::TFFusedBatchNorm::scale ( void  ) const
inline

Definition at line 34 of file TFFusedBatchNorm.h.

34{ return at(1)->node(); } // gamma

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

◆ variance() [1/2]

void moco::TFFusedBatchNorm::variance ( Node *  node)
inline

Definition at line 44 of file TFFusedBatchNorm.h.

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

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

◆ variance() [2/2]

Node * moco::TFFusedBatchNorm::variance ( void  ) const
inline

Definition at line 43 of file TFFusedBatchNorm.h.

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

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

◆ x() [1/2]

void moco::TFFusedBatchNorm::x ( Node *  node)
inline

Definition at line 32 of file TFFusedBatchNorm.h.

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

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

◆ x() [2/2]

Node * moco::TFFusedBatchNorm::x ( void  ) const
inline

Definition at line 31 of file TFFusedBatchNorm.h.

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

References moco::FixedArityNode< 5, TFNodeImpl< TFOpcode::FusedBatchNorm > >::at(), and loco::Use::node().

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