Go to the source code of this file.
|
template<typename T > |
void | magmadnn::math::relu (Tensor< T > *x, Tensor< T > *out) |
|
template<typename T > |
void | magmadnn::math::relu_grad (Tensor< T > *x, Tensor< T > *relu_out, Tensor< T > *grad, Tensor< T > *out) |
|
- Author
- Daniel Nichols
- Version
- 1.0
- Date
- 2019-06-21
- Copyright
- Copyright (c) 2019