diff options
author | Tim Hall <tim.hall@arm.com> | 2021-02-04 21:31:57 +0000 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2021-02-05 11:54:48 +0000 |
commit | 42abec185a95ac8d5f2b32d541f587b3cd306e75 (patch) | |
tree | 7c17eb47cb89a37b9f9d82add61385a5a82c40fe /ethosu/vela/tflite/Model.py | |
parent | 73e843f76dd71e4ab5e07a7616c2c4806ca6ac25 (diff) | |
download | ethos-u-vela-42abec185a95ac8d5f2b32d541f587b3cd306e75.tar.gz |
MLBEDSW-3771: Updated to TF 2.4 flatbuffer schema
Updated tflite loader and mappings from tensorflow 2.3 to tensorflow 2.4
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I55884000ee139baf639bb0377008e0534f72fe94
Diffstat (limited to 'ethosu/vela/tflite/Model.py')
-rw-r--r-- | ethosu/vela/tflite/Model.py | 24 |
1 files changed, 23 insertions, 1 deletions
diff --git a/ethosu/vela/tflite/Model.py b/ethosu/vela/tflite/Model.py index cc9991ba..52aac021 100644 --- a/ethosu/vela/tflite/Model.py +++ b/ethosu/vela/tflite/Model.py @@ -134,7 +134,27 @@ class Model(object): return self._tab.VectorLen(o) return 0 -def ModelStart(builder): builder.StartObject(7) + # Model + def SignatureDefs(self, j): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18)) + if o != 0: + x = self._tab.Vector(o) + x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4 + x = self._tab.Indirect(x) + from .SignatureDef import SignatureDef + obj = SignatureDef() + obj.Init(self._tab.Bytes, x) + return obj + return None + + # Model + def SignatureDefsLength(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18)) + if o != 0: + return self._tab.VectorLen(o) + return 0 + +def ModelStart(builder): builder.StartObject(8) def ModelAddVersion(builder, version): builder.PrependUint32Slot(0, version, 0) def ModelAddOperatorCodes(builder, operatorCodes): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(operatorCodes), 0) def ModelStartOperatorCodesVector(builder, numElems): return builder.StartVector(4, numElems, 4) @@ -147,4 +167,6 @@ def ModelAddMetadataBuffer(builder, metadataBuffer): builder.PrependUOffsetTRela def ModelStartMetadataBufferVector(builder, numElems): return builder.StartVector(4, numElems, 4) def ModelAddMetadata(builder, metadata): builder.PrependUOffsetTRelativeSlot(6, flatbuffers.number_types.UOffsetTFlags.py_type(metadata), 0) def ModelStartMetadataVector(builder, numElems): return builder.StartVector(4, numElems, 4) +def ModelAddSignatureDefs(builder, signatureDefs): builder.PrependUOffsetTRelativeSlot(7, flatbuffers.number_types.UOffsetTFlags.py_type(signatureDefs), 0) +def ModelStartSignatureDefsVector(builder, numElems): return builder.StartVector(4, numElems, 4) def ModelEnd(builder): return builder.EndObject() |