|
template<typename T > |
void | magmadnn::math::relu (Tensor< T > *x, Tensor< T > *out) |
|
template void | magmadnn::math::relu (Tensor< int > *x, Tensor< int > *out) |
|
template void | magmadnn::math::relu (Tensor< float > *x, Tensor< float > *out) |
|
template void | magmadnn::math::relu (Tensor< double > *x, Tensor< double > *out) |
|
template<typename T > |
void | magmadnn::math::relu_grad (Tensor< T > *x, Tensor< T > *relu_out, Tensor< T > *grad, Tensor< T > *out) |
|
template void | magmadnn::math::relu_grad (Tensor< int > *x, Tensor< int > *relu_out, Tensor< int > *grad, Tensor< int > *out) |
|
template void | magmadnn::math::relu_grad (Tensor< float > *x, Tensor< float > *relu_out, Tensor< float > *grad, Tensor< float > *out) |
|
template void | magmadnn::math::relu_grad (Tensor< double > *x, Tensor< double > *relu_out, Tensor< double > *grad, Tensor< double > *out) |
|