|
SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
A TilingConfig describes tiling strategies and optimal tile sizes for inputs, weights, and outputs Tensors. More...
#include <smv_tiling_common.h>
A TilingConfig describes tiling strategies and optimal tile sizes for inputs, weights, and outputs Tensors.
Definition at line 29 of file smv_tiling_common.h.
Public Member Functions | |
| TilingConfig (TensorShape _inputs=TensorShape(), TensorShape _weights=TensorShape(), TensorShape _outputs=TensorShape()) | |
| int | getTotalSize () const |
Public Attributes | |
| TensorShape | inputs |
| TensorShape | weights |
| TensorShape | outputs |
| TilingDims | inputTilingDims |
| TilingDims | weightTilingDims |
| TilingDims | outputTilingDims |
1.8.18