SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Softmax operator on SMV. More...
#include <smv_softmax_op.h>
Softmax operator on SMV.
Definition at line 11 of file smv_softmax_op.h.
Public Member Functions | |
void | tile () override |
void | run () override |
Public Member Functions inherited from smaug::SoftmaxOp< SmvBackend > | |
SoftmaxOp (const std::string &name, Workspace *workspace) | |
void | run () override |
void | run () |
Protected Attributes | |
std::array< TiledTensor, 2 > | tiledTensors |