#include "math/reduce_sum.h"

Include dependency graph for reduce_sum.cpp:

Functions
template<typename T >
void	magmadnn::math::internal::_gemv (T alpha, bool trans_A, Tensor< T > A, Tensor< T > x, T beta, Tensor< T > *out)

template<>
void	magmadnn::math::internal::_gemv (float alpha, bool trans_A, Tensor< float > A, Tensor< float > x, float beta, Tensor< float > *out)

template<>
void	magmadnn::math::internal::_gemv (double alpha, bool trans_A, Tensor< double > A, Tensor< double > x, double beta, Tensor< double > *out)

template<typename T >
void	magmadnn::math::reduce_sum (Tensor< T > x, int axis, Tensor< T > ones, Tensor< T > *out)

template void	magmadnn::math::reduce_sum (Tensor< int > x, int axis, Tensor< int > ones, Tensor< int > *out)

template void	magmadnn::math::reduce_sum (Tensor< float > x, int axis, Tensor< float > ones, Tensor< float > *out)

template void	magmadnn::math::reduce_sum (Tensor< double > x, int axis, Tensor< double > ones, Tensor< double > *out)

Detailed Description

Functions