MagmaDNN  1.0
c++NeuralNetworkFramework
gradients.cpp File Reference
Include dependency graph for gradients.cpp:

Functions

template<typename T >
magmadnn_error_t magmadnn::op::get_grad_table (const std::vector< Operation< T > *> &vars, Operation< T > *graph, GradTable< T > &table)
 
template magmadnn_error_t magmadnn::op::get_grad_table (const std::vector< Operation< int > *> &vars, Operation< int > *graph, GradTable< int > &table)
 
template magmadnn_error_t magmadnn::op::get_grad_table (const std::vector< Operation< float > *> &vars, Operation< float > *graph, GradTable< float > &table)
 
template magmadnn_error_t magmadnn::op::get_grad_table (const std::vector< Operation< double > *> &vars, Operation< double > *graph, GradTable< double > &table)
 
template<typename T >
magmadnn_error_t magmadnn::internal::build_grad (op::Operation< T > *var, op::Operation< T > *graph, op::GradTable< T > &table, Tensor< T > **grad)
 
template magmadnn_error_t magmadnn::internal::build_grad (op::Operation< int > *var, op::Operation< int > *graph, op::GradTable< int > &table, Tensor< int > **grad)
 
template magmadnn_error_t magmadnn::internal::build_grad (op::Operation< float > *var, op::Operation< float > *graph, op::GradTable< float > &table, Tensor< float > **grad)
 
template magmadnn_error_t magmadnn::internal::build_grad (op::Operation< double > *var, op::Operation< double > *graph, op::GradTable< double > &table, Tensor< double > **grad)
 

Detailed Description

Author
Daniel Nichols
Version
0.1
Date
2019-05-17

Function Documentation

◆ build_grad()

template<typename T >
magmadnn_error_t magmadnn::internal::build_grad ( op::Operation< T > *  var,
op::Operation< T > *  graph,
op::GradTable< T > &  table,
Tensor< T > **  grad 
)

Sets the gradients for var.

Template Parameters
Tnumeric
Parameters
varVariable to compute gradients for
graphCompute graph that contains var
tableGradTable to put gradients in
Returns
magmadnn_error_t non-zero on error

◆ get_grad_table()

template<typename T >
magmadnn_error_t magmadnn::op::get_grad_table ( const std::vector< Operation< T > * > &  vars,
Operation< T > *  graph,
GradTable< T > &  table 
)

Given a list of vars and compute graph, fills in a GradTable.

Template Parameters
Tnumeric
Parameters
varsA list of variables whose gradients will be computed
graphHead node of compute graph that contains 'vars'
tableGradTable to be filled in
Returns
magmadnn_error_t non-zero on error