SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Implements the inner product operator. More...
#include <inner_product_op.h>
Implements the inner product operator.
Backend | The Backend specialization of this Operator. |
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 |