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

Implements the inner product operator. More...

#include <inner_product_op.h>

Detailed Description

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

Implements the inner product operator.

Template Parameters
BackendThe Backend specialization of this Operator.

Definition at line 38 of file backend.h.

Public Types

enum  { Inputs, Weights, kNumInputs }
 
enum  { Outputs, kNumOutputs }
 

Public Member Functions

 InnerProductOp (const std::string &name, Workspace *workspace)
 
void setNumOutputs (int _outputs)
 
void run () override
 
bool validate () override
 
TensorShape inferOutputShape () const
 
TensorShape inferWeightsShape () const
 
void createWeightsTensors ()
 
void createOutputTensors ()
 
void createAllTensors () override
 
int getNumOutputs () const
 
int getNumParameters () const override
 
std::vector< TensorBase * > getParameterizableInputs () override
 
bool isSamplingSupported () const override
 
void setSamplingInfo (const SamplingInfo &_sampling) override
 
void run ()
 

Protected Attributes

int numOutputs
 
bool weightsTensorsCreated
 
bool outputTensorsCreated
 
std::string weightsName
 
SamplingInfo sampling
 

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