Compute Library
 23.11
DepthfirstStrategy< TInput, TOutput > Class Template Referenceabstract

#include <pooling_depthfirst.hpp>

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

Public Types

typedef void(* KernelType) (unsigned int n_channels, const TInput *const *, TOutput *const *, bool exclude_padding, unsigned int pad_left, unsigned int pad_top, unsigned int pad_right, unsigned int pad_bottom)
 

Public Member Functions

 DepthfirstStrategy (unsigned int window_rows, unsigned int window_cols, unsigned int stride_rows, unsigned int stride_cols, unsigned int output_rows, unsigned int output_cols)
 
unsigned int get_input_rows () const override
 
unsigned int get_input_cols () const override
 
unsigned int get_output_rows () const override
 
unsigned int get_output_cols () const override
 
virtual KernelType get_kernel (void) const =0
 
- Public Member Functions inherited from IDepthfirstStrategy
virtual ~IDepthfirstStrategy ()=default
 

Detailed Description

template<typename TInput, typename TOutput>
class arm_conv::pooling::DepthfirstStrategy< TInput, TOutput >

Definition at line 39 of file pooling_depthfirst.hpp.

Member Typedef Documentation

◆ KernelType

typedef void(* KernelType(unsigned int n_channels, const TInput *const *, TOutput *const *, bool exclude_padding, unsigned int pad_left, unsigned int pad_top, unsigned int pad_right, unsigned int pad_bottom)

Definition at line 58 of file pooling_depthfirst.hpp.

Constructor & Destructor Documentation

◆ DepthfirstStrategy()

DepthfirstStrategy ( unsigned int  window_rows,
unsigned int  window_cols,
unsigned int  stride_rows,
unsigned int  stride_cols,
unsigned int  output_rows,
unsigned int  output_cols 
)
inline

Definition at line 44 of file pooling_depthfirst.hpp.

47  : input_rows(output_rows + (window_rows - 1) * stride_rows),
48  input_cols(output_cols + (window_cols - 1) * stride_cols),
49  output_rows(output_rows), output_cols(output_cols)
50  {
51  }

Member Function Documentation

◆ get_input_cols()

unsigned int get_input_cols ( ) const
inlineoverridevirtual

Implements IDepthfirstStrategy.

Definition at line 54 of file pooling_depthfirst.hpp.

54 { return input_cols; }

◆ get_input_rows()

unsigned int get_input_rows ( ) const
inlineoverridevirtual

Implements IDepthfirstStrategy.

Definition at line 53 of file pooling_depthfirst.hpp.

53 { return input_rows; }

◆ get_kernel()

virtual KernelType get_kernel ( void  ) const
pure virtual

◆ get_output_cols()

unsigned int get_output_cols ( ) const
inlineoverridevirtual

Implements IDepthfirstStrategy.

Definition at line 56 of file pooling_depthfirst.hpp.

56 { return output_cols; }

◆ get_output_rows()

unsigned int get_output_rows ( ) const
inlineoverridevirtual

Implements IDepthfirstStrategy.

Definition at line 55 of file pooling_depthfirst.hpp.

55 { return output_rows; }

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