MagmaDNN  1.0
c++NeuralNetworkFramework
magmadnn::layer::PoolingLayer< T > Class Template Reference
Inheritance diagram for magmadnn::layer::PoolingLayer< T >:
Collaboration diagram for magmadnn::layer::PoolingLayer< T >:

Public Member Functions

 PoolingLayer (op::Operation< T > *input, const std::vector< unsigned int > &filter_shape={2, 2}, const std::vector< unsigned int > &padding={0, 0}, const std::vector< unsigned int > &strides={1, 1}, pooling_mode mode=MAX_POOL, bool propagate_nan=false)
 
virtual std::vector< op::Operation< T > * > get_weights ()
 
- Public Member Functions inherited from magmadnn::layer::Layer< T >
virtual op::Operation< T > * out ()
 
op::Operation< T > * get_input ()
 
op::Operation< T > * get_output ()
 
std::vector< unsigned int > get_input_shape () const
 
std::vector< unsigned int > get_output_shape () const
 
unsigned int get_input_shape (unsigned int i) const
 
unsigned int get_output_shape (unsigned int i) const
 
void set_name (std::string name)
 
std::string get_name () const
 

Protected Member Functions

void init ()
 
- Protected Member Functions inherited from magmadnn::layer::Layer< T >
 Layer (std::vector< unsigned int > input_shape, op::Operation< T > *input)
 

Protected Attributes

pooling_mode mode
 
bool propagate_nan
 
int filter_h
 
int filter_w
 
int pad_h
 
int pad_w
 
int stride_h
 
int stride_w
 
- Protected Attributes inherited from magmadnn::layer::Layer< T >
std::vector< unsigned int > input_shape
 
std::vector< unsigned int > output_shape
 
op::Operation< T > * input
 
op::Operation< T > * output
 
std::string name
 

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