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:

Public Member Functions

Protected Member Functions