Compute Library
 22.11
GenericDepthfirstWrapper< TInput, TOutput, OutputStage > Class Template Reference

#include <pooling_depthfirst_generic.hpp>

Collaboration diagram for GenericDepthfirstWrapper< TInput, TOutput, OutputStage >:
[legend]

Public Member Functions

 GenericDepthfirstWrapper (const StratType *strat, const PoolingArgs &args)
 
unsigned int get_input_rows (void) const override
 
unsigned int get_input_cols (void) const override
 
unsigned int get_output_rows (void) const override
 
unsigned int get_output_cols (void) const override
 
StratType::KernelType get_kernel (void) const
 
- Public Member Functions inherited from IDepthfirstStrategy
virtual ~IDepthfirstStrategy ()=default
 

Detailed Description

template<typename TInput, typename TOutput, typename OutputStage>
class arm_conv::pooling::GenericDepthfirstWrapper< TInput, TOutput, OutputStage >

Definition at line 113 of file pooling_depthfirst_generic.hpp.

Constructor & Destructor Documentation

◆ GenericDepthfirstWrapper()

GenericDepthfirstWrapper ( const StratType strat,
const PoolingArgs &  args 
)
inline

Definition at line 121 of file pooling_depthfirst_generic.hpp.

122  : m_strat(strat), window_rows(args.pool_window.rows), window_cols(args.pool_window.cols)
123  {
124  }

Member Function Documentation

◆ get_input_cols()

unsigned int get_input_cols ( void  ) const
inlineoverridevirtual

Implements IDepthfirstStrategy.

Definition at line 127 of file pooling_depthfirst_generic.hpp.

127 { return window_cols; }

◆ get_input_rows()

unsigned int get_input_rows ( void  ) const
inlineoverridevirtual

Implements IDepthfirstStrategy.

Definition at line 126 of file pooling_depthfirst_generic.hpp.

126 { return window_rows; }

◆ get_kernel()

StratType::KernelType get_kernel ( void  ) const
inline

Definition at line 131 of file pooling_depthfirst_generic.hpp.

131 { return m_strat->get_kernel(); }

◆ get_output_cols()

unsigned int get_output_cols ( void  ) const
inlineoverridevirtual

Implements IDepthfirstStrategy.

Definition at line 129 of file pooling_depthfirst_generic.hpp.

129 { return 1; }

◆ get_output_rows()

unsigned int get_output_rows ( void  ) const
inlineoverridevirtual

Implements IDepthfirstStrategy.

Definition at line 128 of file pooling_depthfirst_generic.hpp.

128 { return 1; }

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