#include "math/crossentropy.h"
Functions | |
template<typename T > | |
void | magmadnn::math::crossentropy (Tensor< T > *predicted, Tensor< T > *ground_truth, Tensor< T > *out) |
template void | magmadnn::math::crossentropy (Tensor< int > *predicted, Tensor< int > *ground_truth, Tensor< int > *out) |
template void | magmadnn::math::crossentropy (Tensor< float > *predicted, Tensor< float > *ground_truth, Tensor< float > *out) |
template void | magmadnn::math::crossentropy (Tensor< double > *predicted, Tensor< double > *ground_truth, Tensor< double > *out) |