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

T

Numeric

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:

• include/compute/gradtable.h
• src/compute/gradtable.cpp