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

Implements the scaled exponential linear unit function. More...

#include <elu_op.h>

Detailed Description

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

Implements the scaled exponential linear unit function.

Defined as: lambda * elu(input).

Template Parameters
BackendThe Backend specialization of this Operator.

Definition at line 58 of file backend.h.

Public Member Functions

 SeluOp (const std::string &name, Workspace *workspace)
 
void run () override
 
void setLambda (float _lambda)
 
float getLambda () const
 
void run ()
 

Protected Attributes

float lambda
 

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