|
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 |
1.8.18