Inheritance diagram for magmadnn::layer::Conv2dLayer< T >:

Collaboration diagram for magmadnn::layer::Conv2dLayer< T >:

Public Member Functions
	Conv2dLayer (op::Operation< T > *input, const std::vector< unsigned int > &filter_shape={3, 3}, int out_channels=1, const std::vector< unsigned int > &padding={0, 0}, const std::vector< unsigned int > &strides={1, 1}, const std::vector< unsigned int > &dilation_rates={1, 1}, bool use_cross_correlation=true, bool use_bias=false, tensor_filler_t< T > filter_initializer={GLOROT,{0.0, 0.2f}}, tensor_filler_t< T > bias_initializer={GLOROT,{0.0, 0.2f}})

virtual std::vector< op::Operation< T > * >	get_weights ()

op::Operation< T > *	get_filter ()

op::Operation< T > *	get_bias ()

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 (const std::vector< unsigned int > &filter_shape)

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

Protected Attributes
Tensor< T > *	filter_tensor

Tensor< T > *	bias_tensor

op::Operation< T > *	filter

op::Operation< T > *	bias

int	in_channels

int	out_channels

bool	use_cross_correlation

bool	use_bias

int	pad_h

int	pad_w

int	stride_h

int	stride_w

int	dilation_h

int	dilation_w

tensor_filler_t< T >	filter_initializer

tensor_filler_t< T >	bias_initializer

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