SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Implements the hard tanh operator, which bounds the min and max value of the tanh operator. More...
#include <tanh_op.h>
Implements the hard tanh operator, which bounds the min and max value of the tanh operator.
Backend | The Backend specialization of this Operator. |
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 |