lbann::dnn_lib Namespace Reference

Classes
class	data_parallel_layer_tensor_manager
class	entrywise_layer_tensor_manager

Functions
template<typename ScalarT , typename TensorDescT , typename DataT , El::Device D>
void	softmax_forward (ScalarT const &alpha_in, TensorDescT const &xDesc, El::Matrix< DataT, D > const &x, ScalarT const &beta_in, TensorDescT const &yDesc, El::Matrix< DataT, D > &y, El::SyncInfo< D > const &si, softmax_mode mode, softmax_alg alg=softmax_alg::ACCURATE)
template<typename ScalarT , typename TensorDescT , typename DataT , El::Device D>
void	softmax_forward (ScalarT const &alpha_in, TensorDescT const &xDesc, El::Matrix< DataT, D > const &x, ScalarT const &beta_in, TensorDescT const &yDesc, El::Matrix< DataT, D > &y, softmax_mode mode, softmax_alg alg=softmax_alg::ACCURATE)
template<typename ScalarT , typename TensorDescT , typename DataT , El::Device D>
void	softmax_backward (ScalarT const &alpha_in, TensorDescT const &yDesc, El::Matrix< DataT, D > const &y, TensorDescT const &dyDesc, El::Matrix< DataT, D > const &dy, ScalarT const &beta_in, TensorDescT const &dxDesc, El::Matrix< DataT, D > &dx, El::SyncInfo< D > const &si, softmax_mode mode, softmax_alg alg=softmax_alg::ACCURATE)
template<typename ScalarT , typename TensorDescT , typename DataT , El::Device D>
void	softmax_backward (ScalarT const &alpha_in, TensorDescT const &yDesc, El::Matrix< DataT, D > const &y, TensorDescT const &dyDesc, El::Matrix< DataT, D > const &dy, ScalarT const &beta_in, TensorDescT const &dxDesc, El::Matrix< DataT, D > &dx, softmax_mode mode, softmax_alg alg=softmax_alg::ACCURATE)

Function Documentation

softmax_backward() [1/2]

template<typename ScalarT , typename TensorDescT , typename DataT , El::Device D>

void lbann::dnn_lib::softmax_backward	(	ScalarT const &	alpha_in,
		TensorDescT const &	yDesc,
		El::Matrix< DataT, D > const &	y,
		TensorDescT const &	dyDesc,
		El::Matrix< DataT, D > const &	dy,
		ScalarT const &	beta_in,
		TensorDescT const &	dxDesc,
		El::Matrix< DataT, D > &	dx,
		El::SyncInfo< D > const &	si,
		softmax_mode	mode,
		softmax_alg	alg = softmax_alg::ACCURATE
	)

Definition at line 98 of file utils/dnn_lib/softmax.hpp.

Here is the call graph for this function:

Here is the caller graph for this function:

softmax_backward() [2/2]

template<typename ScalarT , typename TensorDescT , typename DataT , El::Device D>

void lbann::dnn_lib::softmax_backward	(	ScalarT const &	alpha_in,
		TensorDescT const &	yDesc,
		El::Matrix< DataT, D > const &	y,
		TensorDescT const &	dyDesc,
		El::Matrix< DataT, D > const &	dy,
		ScalarT const &	beta_in,
		TensorDescT const &	dxDesc,
		El::Matrix< DataT, D > &	dx,
		softmax_mode	mode,
		softmax_alg	alg = softmax_alg::ACCURATE
	)

Definition at line 130 of file utils/dnn_lib/softmax.hpp.

Here is the call graph for this function:

Here is the caller graph for this function:

softmax_forward() [1/2]

template<typename ScalarT , typename TensorDescT , typename DataT , El::Device D>

void lbann::dnn_lib::softmax_forward	(	ScalarT const &	alpha_in,
		TensorDescT const &	xDesc,
		El::Matrix< DataT, D > const &	x,
		ScalarT const &	beta_in,
		TensorDescT const &	yDesc,
		El::Matrix< DataT, D > &	y,
		El::SyncInfo< D > const &	si,
		softmax_mode	mode,
		softmax_alg	alg = softmax_alg::ACCURATE
	)

Definition at line 52 of file utils/dnn_lib/softmax.hpp.

Here is the call graph for this function:

Here is the caller graph for this function:

softmax_forward() [2/2]

template<typename ScalarT , typename TensorDescT , typename DataT , El::Device D>

void lbann::dnn_lib::softmax_forward	(	ScalarT const &	alpha_in,
		TensorDescT const &	xDesc,
		El::Matrix< DataT, D > const &	x,
		ScalarT const &	beta_in,
		TensorDescT const &	yDesc,
		El::Matrix< DataT, D > &	y,
		softmax_mode	mode,
		softmax_alg	alg = softmax_alg::ACCURATE
	)

Definition at line 76 of file utils/dnn_lib/softmax.hpp.

Here is the call graph for this function:

Here is the caller graph for this function: