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

Workspace is the container and owner of all Tensors and Operators in the Network. More...

#include <workspace.h>

Detailed Description

Workspace is the container and owner of all Tensors and Operators in the Network.

Every Tensor/Operator that is created must be added to a Workspace (and in general, there is only one Workspace).

Definition at line 17 of file workspace.h.

Public Member Functions

TensoraddTensor (Tensor *tensor)
 
void addTiledTensor (TiledTensor &tiledTensor)
 
TensorgetTensor (const std::string &name) const
 
TensorgetTensor (Operator *op) const
 

Protected Attributes

std::map< std::string, TensorBase * > tensors
 

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