Go to the source code of this file.
|
template<typename T > |
void | magmadnn::math::dropout (Tensor< T > *x, Tensor< T > *out, Tensor< T > *mask, float dropout_rate) |
|
template<typename T > |
void | magmadnn::math::dropout_grad (Tensor< T > *grad, Tensor< T > *out, Tensor< T > *mask) |
|
- Author
- Sedrick Keh
- Version
- 0.1
- Date
- 2019-06-28
- Copyright
- Copyright (c) 2019