MagmaDNN  1.0
c++NeuralNetworkFramework
magmadnn::layer::Conv2dLayer< T > Class Template Reference
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: