gradients.cpp File Reference

#include "compute/gradients.h"

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

Copyright

Copyright (c) 2019

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

T

numeric

Parameters

var	Variable to compute gradients for
graph	Compute graph that contains var
table	GradTable 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

T

numeric

Parameters

vars	A list of variables whose gradients will be computed
graph	Head node of compute graph that contains 'vars'
table	GradTable to be filled in

Returns

magmadnn_error_t non-zero on error