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 |