From 22f74e1c39572f084ad05cc2f208446fd2f50138 Mon Sep 17 00:00:00 2001 From: Michael McGeagh Date: Fri, 7 Aug 2020 16:21:03 +0100 Subject: MLBEDSW-2383 Preserve previous metadata The input tflite file potentially has metadata attached to it, which was lost when writing the vela optimised tflite file out. This patch preserves any metadata found. Signed-off-by: Michael McGeagh Change-Id: I7b4e941696d21b81802fd4398cd405323778bedf --- ethosu/vela/nn_graph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela/nn_graph.py') diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index 4a2855b2..bfab2270 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -502,7 +502,7 @@ class Graph: self.name = name self.batch_size = batch_size self.subgraphs = [] - + self.metadata = [] self.memory_used = {} self.bits_per_element = {} self.total_size = {} -- cgit v1.2.1