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

#include <TFAvgPool.h>

Collaboration diagram for moco::TFAvgPool:

Public Member Functions

 TFAvgPool ()=default
 
Nodevalue (void) const
 
void value (Node *node)
 
const TFDataLayoutdata_layout (void) const
 
void data_layout (const TFDataLayout &data_layout)
 
const TFPaddingpadding (void) const
 
void padding (const TFPadding &padding)
 
const std::vector< int64_t > & ksize (void) const
 
void ksize (const std::vector< int64_t > &ksize)
 
const std::vector< int64_t > & strides (void) const
 
void strides (const std::vector< int64_t > &strides)
 
- Public Member Functions inherited from moco::FixedArityNode< 1, TFNodeImpl< TFOpcode::AvgPool > >
 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::AvgPool >
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< 1, TFNodeImpl< TFOpcode::AvgPool > >
loco::Useat (unsigned n) const
 

Detailed Description

Note
TFAvgPool corresponds to the following GraphDef

Definition at line 70 of file TFAvgPool.h.

Constructor & Destructor Documentation

◆ TFAvgPool()

moco::TFAvgPool::TFAvgPool ( )
default

Member Function Documentation

◆ data_layout() [1/2]

void moco::TFAvgPool::data_layout ( const TFDataLayout data_layout)
inline

Definition at line 81 of file TFAvgPool.h.

81{ _data_layout = data_layout; }
const TFDataLayout & data_layout(void) const
Definition TFAvgPool.h:80

References data_layout().

Referenced by data_layout().

◆ data_layout() [2/2]

const TFDataLayout & moco::TFAvgPool::data_layout ( void  ) const
inline

Definition at line 80 of file TFAvgPool.h.

80{ return _data_layout; }

◆ ksize() [1/2]

void moco::TFAvgPool::ksize ( const std::vector< int64_t > &  ksize)
inline

Definition at line 87 of file TFAvgPool.h.

87{ _ksize = ksize; }
const std::vector< int64_t > & ksize(void) const
Definition TFAvgPool.h:86

References ksize().

Referenced by ksize().

◆ ksize() [2/2]

const std::vector< int64_t > & moco::TFAvgPool::ksize ( void  ) const
inline

Definition at line 86 of file TFAvgPool.h.

86{ return _ksize; }

◆ padding() [1/2]

void moco::TFAvgPool::padding ( const TFPadding padding)
inline

Definition at line 84 of file TFAvgPool.h.

84{ _padding = padding; }
const TFPadding & padding(void) const
Definition TFAvgPool.h:83

References padding().

Referenced by padding().

◆ padding() [2/2]

const TFPadding & moco::TFAvgPool::padding ( void  ) const
inline

Definition at line 83 of file TFAvgPool.h.

83{ return _padding; }

◆ strides() [1/2]

void moco::TFAvgPool::strides ( const std::vector< int64_t > &  strides)
inline

Definition at line 90 of file TFAvgPool.h.

90{ _strides = strides; }
const std::vector< int64_t > & strides(void) const
Definition TFAvgPool.h:89

References strides().

Referenced by strides().

◆ strides() [2/2]

const std::vector< int64_t > & moco::TFAvgPool::strides ( void  ) const
inline

Definition at line 89 of file TFAvgPool.h.

89{ return _strides; }

◆ value() [1/2]

void moco::TFAvgPool::value ( Node node)
inline

Definition at line 77 of file TFAvgPool.h.

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

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

◆ value() [2/2]

Node * moco::TFAvgPool::value ( void  ) const
inline

Definition at line 76 of file TFAvgPool.h.

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

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


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