aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/nn_graph.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/nn_graph.py')
-rw-r--r--ethosu/vela/nn_graph.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index ed2ab322..ea35c087 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -485,6 +485,7 @@ class Graph:
self.bits_per_element = {}
self.total_size = {}
self.total_elements = {}
+ self.weight_cache = None # See CompressedWeightCache
def get_root_subgraph(self):
return self.subgraphs[0]