MagmaDNN  1.0
c++NeuralNetworkFramework
Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
magmadnn::op::Conv2DForwardOp< T > Class Template Reference
Inheritance diagram for magmadnn::op::Conv2DForwardOp< T >:
Inheritance graph
[legend]
Collaboration diagram for magmadnn::op::Conv2DForwardOp< T >:
Collaboration graph
[legend]

Public Member Functions

 Conv2DForwardOp (Operation< T > *input, Operation< T > *filter, int pad_h=0, int pad_w=0, int vertical_stride=1, int horizontal_stride=1, int dilation_h=1, int dilation_w=1, bool use_cross_correlation=true, bool needs_grad=true)
 
std::string to_string ()
 
- Public Member Functions inherited from magmadnn::op::Operation< T >
 Operation ()
 
 Operation (std::vector< Operation< T > *> inputs, bool needs_grad=true)
 
virtual std::vector< unsigned int > get_output_shape () const
 
virtual unsigned int get_output_shape (unsigned int idx) const
 
virtual unsigned int get_output_size () const
 
virtual memory_t get_memory_type () const
 
virtual Tensor< T > * eval (bool recompute=true)
 
virtual void reset ()
 
virtual Tensor< T > * grad (Operation< T > *consumer, Operation< T > *var, Tensor< T > *grad, bool recompute=true)
 
virtual void add_consumer (Operation< T > *consumer)
 
virtual std::vector< Operation< T > * > get_consumers ()
 
virtual std::vector< Operation< T > * > get_inputs ()
 
virtual Tensor< T > * get_output_tensor ()
 
virtual Tensor< T > * get_grad_tensor (Operation< T > *wrt)
 
virtual std::string get_name ()
 

Protected Member Functions

Tensor< T > * _eval (bool recompute)
 
Tensor< T > * _grad (Operation< T > *consumer, Operation< T > *var, Tensor< T > *grad)
 
void init_settings ()
 
void calculate_and_set_output_shape ()
 

Protected Attributes

Operation< T > * input
 
Operation< T > * filter
 
Tensor< T > * input_tensor
 
Tensor< T > * filter_tensor
 
int pad_h
 
int pad_w
 
int vertical_stride
 
int horizontal_stride
 
int dilation_h
 
int dilation_w
 
bool use_cross_correlation
 
- Protected Attributes inherited from magmadnn::op::Operation< T >
std::vector< Operation< T > * > inputs
 
std::vector< Operation< T > * > consumers
 
std::vector< unsigned int > output_shape
 
memory_t mem_type
 
std::map< uintptr_t, Tensor< T > * > _grad_cache
 
std::string name = "DefaultOpName"
 
Tensor< T > * output_tensor
 
bool needs_grad
 
bool has_been_computed
 
bool has_grad_been_computed
 

Member Function Documentation

◆ _eval()

template<typename T >
Tensor< T > * magmadnn::op::Conv2DForwardOp< T >::_eval ( bool  recompute)
protectedvirtual

Sets this->output_tensor to the value of this operation

Returns
Tensor<T>* the evaluated tensor

Implements magmadnn::op::Operation< T >.

Here is the caller graph for this function:

◆ _grad()

template<typename T >
Tensor< T > * magmadnn::op::Conv2DForwardOp< T >::_grad ( Operation< T > *  consumer,
Operation< T > *  var,
Tensor< T > *  grad 
)
protectedvirtual

Computes the gradient of this operation wrt the output of consumer.

Parameters
consumer
var
grad
recompute
Returns
Tensor<T>*

Implements magmadnn::op::Operation< T >.

Here is the caller graph for this function:

◆ to_string()

template<typename T>
std::string magmadnn::op::Conv2DForwardOp< T >::to_string ( )
inlinevirtual

string form of the given operation. Expands on input.

Returns
std::string

Implements magmadnn::op::Operation< T >.

Here is the call graph for this function:
The documentation for this class was generated from the following files: