utils/dnn_lib/softmax.hpp File Reference

#include "lbann_config.hpp"
#include "lbann/utils/dnn_lib/openmp/softmax.hpp"
#include "lbann/utils/sync_info_helpers.hpp"

Include dependency graph for utils/dnn_lib/softmax.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	lbann
	lbann::dnn_lib

Functions
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)
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)
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)
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)