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
diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py
index 9346b76..bf3fe95 100644
--- a/ethosu/vela/tflite_reader.py
+++ b/ethosu/vela/tflite_reader.py
@@ -245,6 +245,14 @@
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