SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
Public Member Functions | Protected Attributes | List of all members
smaug::EluOp< Backend > Class Template Reference

Implements the exponential linear unit function. More...

#include <elu_op.h>

Detailed Description

template<typename Backend>
class smaug::EluOp< Backend >

Implements the exponential linear unit function.

Defined as: if input > 0, alpha * exp(input - 1), else input.

Template Parameters
BackendThe Backend specialization of this Operator.

Definition at line 57 of file backend.h.

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
 

The documentation for this class was generated from the following files: