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

Implements the hard tanh operator, which bounds the min and max value of the tanh operator. More...

#include <tanh_op.h>

Detailed Description

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

Implements the hard tanh operator, which bounds the min and max value of the tanh operator.

Template Parameters
BackendThe Backend specialization of this Operator.

Definition at line 60 of file backend.h.

Public Member Functions

 HardTanhOp (const std::string &name, Workspace *workspace, float _min=-1, float _max=1)
 
void run () override
 
void setMin (float _min)
 
void setMax (float _max)
 
float getMin () const
 
float getMax () const
 
void run ()
 

Protected Attributes

float min
 
float max
 

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