SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
Public Member Functions | Protected Attributes | List of all members
smaug::SmvSoftmaxOp Class Reference

Softmax operator on SMV. More...

#include <smv_softmax_op.h>

Inheritance diagram for smaug::SmvSoftmaxOp:
smaug::SoftmaxOp< SmvBackend >

Detailed Description

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
 

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