SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Implements the softmax operator. More...
#include <softmax_op.h>
Implements the softmax operator.
Backend | The Backend specialization of this Operator. |
Public Member Functions | |
SoftmaxOp (const std::string &name, Workspace *workspace) | |
void | run () override |
void | run () |