#include "math/dropout.h"

Include dependency graph for dropout.cpp:

Functions
template<typename T >
void	magmadnn::math::dropout (Tensor< T > x, Tensor< T > out, Tensor< T > *mask, float dropout_rate)

template void	magmadnn::math::dropout (Tensor< int > x, Tensor< int > out, Tensor< int > *mask, float dropout_rate)

template void	magmadnn::math::dropout (Tensor< float > x, Tensor< float > out, Tensor< float > *mask, float dropout_rate)

template void	magmadnn::math::dropout (Tensor< double > x, Tensor< double > out, Tensor< double > *mask, float dropout_rate)

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

template void	magmadnn::math::dropout_grad (Tensor< int > grad, Tensor< int > out, Tensor< int > *mask)

template void	magmadnn::math::dropout_grad (Tensor< float > grad, Tensor< float > out, Tensor< float > *mask)

template void	magmadnn::math::dropout_grad (Tensor< double > grad, Tensor< double > out, Tensor< double > *mask)

Detailed Description

Functions