SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Elementwise scaled exponential linear unit on SMV. More...
#include <smv_elu_op.h>
Elementwise scaled exponential linear unit on SMV.
Definition at line 23 of file smv_elu_op.h.
Public Member Functions | |
void | tile () override |
void | run () override |
Public Member Functions inherited from smaug::SeluOp< SmvBackend > | |
SeluOp (const std::string &name, Workspace *workspace) | |
void | run () override |
void | run () |
void | setLambda (float _lambda) |
float | getLambda () const |
Protected Attributes | |
std::array< TiledTensor, 2 > | tiledTensors |
Protected Attributes inherited from smaug::SeluOp< SmvBackend > | |
float | lambda |