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/tflite_reader.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/tflite_reader.py')
-rw-r--r-- | ethosu/vela/tflite_reader.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py index 9346b760..bf3fe950 100644 --- a/ethosu/vela/tflite_reader.py +++ b/ethosu/vela/tflite_reader.py @@ -245,6 +245,14 @@ class TFLiteGraph: sg.output_tensors = tflite_sg.outputs self.nng.subgraphs.append(sg) + # Preserve the original metadata + for idx in range(model.MetadataLength()): + meta = model.Metadata(idx) + name = meta.Name() + if name is not None: + buf_data = self.buffers[meta.Buffer()] + self.nng.metadata.append((name, buf_data)) + def parse_buffer(self, buf_data): if buf_data.DataLength() == 0: return None |