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 |