ONE - On-device Neural Engine
Loading...
Searching...
No Matches
moco::TFMean Class Referencefinal

#include <TFMean.h>

Collaboration diagram for moco::TFMean:

Public Member Functions

 TFMean ()=default
 
Nodeinput (void) const
 
void input (Node *node)
 
Nodereduction_indices (void) const
 
void reduction_indices (Node *node)
 
bool keep_dims (void) const
 
void keep_dims (bool keep_dims)
 
- Public Member Functions inherited from moco::FixedArityNode< 2, TFNodeImpl< TFOpcode::Mean > >
 FixedArityNode ()
 
virtual ~FixedArityNode ()=default
 
unsigned arity (void) const final
 
loco::Nodearg (uint32_t n) const final
 
void drop (void) final
 
- Public Member Functions inherited from moco::TFNodeImpl< TFOpcode::Mean >
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::Dialectdialect (void) const final
 Return "Dialect" identifier that this node belongs to.
 
template<typename T >
accept (TFNodeVisitorBase< T > *) const
 
template<typename 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 ()
 
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 moco::FixedArityNode< 2, TFNodeImpl< TFOpcode::Mean > >
loco::Useat (unsigned n) const
 

Detailed Description

Note
TFMean corresponds to the following GraphDef

Definition at line 49 of file TFMean.h.

Constructor & Destructor Documentation

◆ TFMean()

moco::TFMean::TFMean ( )
default

Member Function Documentation

◆ input() [1/2]

void moco::TFMean::input ( Node node)
inline

Definition at line 56 of file TFMean.h.

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

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

◆ input() [2/2]

Node * moco::TFMean::input ( void  ) const
inline

Definition at line 55 of file TFMean.h.

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

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

◆ keep_dims() [1/2]

void moco::TFMean::keep_dims ( bool  keep_dims)
inline

Definition at line 63 of file TFMean.h.

63{ _keep_dims = keep_dims; }
bool keep_dims(void) const
Definition TFMean.h:62

References keep_dims().

Referenced by keep_dims().

◆ keep_dims() [2/2]

bool moco::TFMean::keep_dims ( void  ) const
inline

Definition at line 62 of file TFMean.h.

62{ return _keep_dims; }

◆ reduction_indices() [1/2]

void moco::TFMean::reduction_indices ( Node node)
inline

Definition at line 59 of file TFMean.h.

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

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

◆ reduction_indices() [2/2]

Node * moco::TFMean::reduction_indices ( void  ) const
inline

Definition at line 58 of file TFMean.h.

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

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


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