MagmaDNN  1.0
c++NeuralNetworkFramework
magmadnn::op::GradTable< T > Class Template Reference

#include <gradtable.h>

Collaboration diagram for magmadnn::op::GradTable< T >:

Public Member Functions

 GradTable ()
 
unsigned int get_size ()
 
Tensor< T > * get (Operation< T > *var)
 
void set (Operation< T > *var, Tensor< T > *grad)
 
void clear ()
 

Protected Attributes

std::map< Operation< T > *, Tensor< T > *> _table
 
std::map< Operation< T > *, Tensor< T > * >::iterator tmp_map_iterator
 

Detailed Description

template<typename T>
class magmadnn::op::GradTable< T >

GradTable class.

Template Parameters
TNumeric

Constructor & Destructor Documentation

◆ GradTable()

template<typename T >
magmadnn::op::GradTable< T >::GradTable ( )

Constructs a new grad table.

Member Function Documentation

◆ clear()

template<typename T >
void magmadnn::op::GradTable< T >::clear ( )

Removes all entries.

◆ get()

template<typename T >
Tensor< T > * magmadnn::op::GradTable< T >::get ( Operation< T > *  var)

Takes a variable and returns the tree to compute its gradient.

Parameters
var
Returns
Operation<T>*

◆ get_size()

template<typename T >
unsigned int magmadnn::op::GradTable< T >::get_size ( )

returns the size of this grad table.

Returns
unsigned int the size of this grad table

◆ set()

template<typename T >
void magmadnn::op::GradTable< T >::set ( Operation< T > *  var,
Tensor< T > *  grad 
)

Sets var's gradient to grad.

Parameters
var
grad

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