SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Workspace is the container and owner of all Tensors and Operators in the Network. More...
#include <workspace.h>
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 | |
Tensor * | addTensor (Tensor *tensor) |
void | addTiledTensor (TiledTensor &tiledTensor) |
Tensor * | getTensor (const std::string &name) const |
Tensor * | getTensor (Operator *op) const |
Protected Attributes | |
std::map< std::string, TensorBase * > | tensors |