|
ZigZag - Deep Learning Hardware Design Space Exploration
This repository presents the novel version of our tried-and-tested hardware Architecture-Mapping Design Space Exploration (DSE) Framework for Deep Learning (DL) accelerators. ZigZag bridges the gap between algorithmic DL decisions and their acceleration cost on specialized accelerators through a fast and accurate hardware cost estimation.
|
Suggested LayerDims to be unrolled for every OADimension. More...


Public Member Functions | |
| def | __init__ (self, dict[OADimension, set[LayerDim]] data) |
| def | clear_invalid_hits (self, list[LayerDim] valid_layer_dims) |
| Check the hints at all contained OADimension. More... | |
| def | complete_with_defaults (self, dict[OADimension, int] oa_dim_sizes, list[LayerDim] layer_dims) |
For all OADimensions in oa_dim_sizes that are not already in this SpatialMappingHint, fill the hints for that OADimension with all LayerDims from layer_dims. More... | |
| def | __getitem__ (self, OADimension key) |
Public Member Functions inherited from LayerAttribute | |
| def | __init__ (self, Any data) |
| int | __len__ (self) |
| Iterator[Any] | __iter__ (self) |
| def | __getitem__ (self, Any key) |
| bool | __contains__ (self, Any key) |
| def | __str__ (self) |
| def | __repr__ (self) |
| Any | __jsonrepr__ (self) |
| def | __eq__ (self, object other) |
| def | __hash__ (self) |
Static Public Member Functions | |
| "SpatialMappingHint" | empty () |
Public Attributes | |
| data | |
Public Attributes inherited from LayerAttribute | |
| data | |
Suggested LayerDims to be unrolled for every OADimension.
| def __init__ | ( | self, | |
| dict[OADimension, set[LayerDim]] | data | ||
| ) |
| def __getitem__ | ( | self, | |
| OADimension | key | ||
| ) |
| def clear_invalid_hits | ( | self, | |
| list[LayerDim] | valid_layer_dims | ||
| ) |
Check the hints at all contained OADimension.
If the OADimension doesn't contain a single LayerDim that is also present in the given list valid_layer_dims, remove the OADimension from this instance.
| def complete_with_defaults | ( | self, | |
| dict[OADimension, int] | oa_dim_sizes, | ||
| list[LayerDim] | layer_dims | ||
| ) |
For all OADimensions in oa_dim_sizes that are not already in this SpatialMappingHint, fill the hints for that OADimension with all LayerDims from layer_dims.
|
static |
| data |