|
SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Implements the exponential linear unit function. More...
#include <elu_op.h>
Implements the exponential linear unit function.
Defined as: if input > 0, alpha * exp(input - 1), else input.
| Backend | The Backend specialization of this Operator. |
Public Member Functions | |
| EluOp (const std::string &name, Workspace *workspace, float _alpha=0.1) | |
| void | run () override |
| void | setAlpha (float _alpha) |
| float | getAlpha () const |
| void | run () |
Protected Attributes | |
| float | alpha |
1.8.18