SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
Public Member Functions | Protected Attributes | List of all members
smaug::DataOp< Backend > Class Template Reference

Exposes a Tensor as its only output. More...

#include <data_op.h>

Detailed Description

template<typename Backend>
class smaug::DataOp< Backend >

Exposes a Tensor as its only output.

This is the only operator that is not expected to have any inputs. Its existence is to maintain the abstraction that the input to all other operators is always provided by another Operator.

Template Parameters
BackendThe Backend specialization of this Operator.

Definition at line 34 of file backend.h.

Public Member Functions

 DataOp (const std::string &name, Workspace *workspace)
 
void setData (Tensor *_data)
 
void run () override
 
bool validate () override
 
void createAllTensors () override
 

Protected Attributes

Tensordata
 

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