diff options
author | Michael McGeagh <michael.mcgeagh@arm.com> | 2020-08-07 16:21:03 +0100 |
---|---|---|
committer | Fredrik Knutsson <fredrik.knutsson.hunnebo@gmail.com> | 2020-08-12 06:25:23 +0000 |
commit | 22f74e1c39572f084ad05cc2f208446fd2f50138 (patch) | |
tree | 96d602818de364e2297fb67c2f00fc1245ba74f7 /ethosu/vela/nn_graph.py | |
parent | e99b893beaa1b95ee86d51a613f208f9f4edf150 (diff) | |
download | ethos-u-vela-22f74e1c39572f084ad05cc2f208446fd2f50138.tar.gz |
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 <michael.mcgeagh@arm.com>
Change-Id: I7b4e941696d21b81802fd4398cd405323778bedf
Diffstat (limited to 'ethosu/vela/nn_graph.py')
-rw-r--r-- | ethosu/vela/nn_graph.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 = {} |