#include "math/softmax.h"

Include dependency graph for softmax.cpp:

Functions
template<typename T >
void	magmadnn::math::softmax (Tensor< T > x, Tensor< T > out)

template void	magmadnn::math::softmax (Tensor< int > x, Tensor< int > out)

template void	magmadnn::math::softmax (Tensor< float > x, Tensor< float > out)

template void	magmadnn::math::softmax (Tensor< double > x, Tensor< double > out)

template<typename T >
void	magmadnn::math::softmax_grad (Tensor< T > softmax, Tensor< T > grad, Tensor< T > *out)

template void	magmadnn::math::softmax_grad (Tensor< int > softmax, Tensor< int > grad, Tensor< int > *out)

template void	magmadnn::math::softmax_grad (Tensor< float > softmax, Tensor< float > grad, Tensor< float > *out)

template void	magmadnn::math::softmax_grad (Tensor< double > softmax, Tensor< double > grad, Tensor< double > *out)

Detailed Description

Functions