aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite/BatchMatMulOptions.py
diff options
context:
space:
mode:
authorDwight Lidman <dwight.lidman@arm.com>2021-07-19 13:43:05 +0200
committertim.hall <tim.hall@arm.com>2021-08-05 12:46:35 +0000
commit8a12da19e247750770bd0c2465d3690c1fcd7316 (patch)
tree7d49a80088464b4d62ac01217f9067a91b6e67f0 /ethosu/vela/tflite/BatchMatMulOptions.py
parent8ae292931aa9b4677092f51a46c0fb07a8807f89 (diff)
downloadethos-u-vela-8a12da19e247750770bd0c2465d3690c1fcd7316.tar.gz
MLBEDSW-4854: Update to TensorFlow 2.5
This commit updates the flatbuffers generated code to comply with TensorFlow 2.5, as well as stripping away some legacy code. Signed-off-by: Dwight Lidman <dwight.lidman@arm.com> Change-Id: I01fe47ec2bde6e78fdde21ee1bc0a71f560c53ae
Diffstat (limited to 'ethosu/vela/tflite/BatchMatMulOptions.py')
-rw-r--r--ethosu/vela/tflite/BatchMatMulOptions.py14
1 files changed, 13 insertions, 1 deletions
diff --git a/ethosu/vela/tflite/BatchMatMulOptions.py b/ethosu/vela/tflite/BatchMatMulOptions.py
index 5be66835..771c36e2 100644
--- a/ethosu/vela/tflite/BatchMatMulOptions.py
+++ b/ethosu/vela/tflite/BatchMatMulOptions.py
@@ -14,6 +14,10 @@ class BatchMatMulOptions(object):
x.Init(buf, n + offset)
return x
+ @classmethod
+ def BatchMatMulOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+ return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
+
# BatchMatMulOptions
def Init(self, buf, pos):
self._tab = flatbuffers.table.Table(buf, pos)
@@ -32,7 +36,15 @@ class BatchMatMulOptions(object):
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
return False
-def BatchMatMulOptionsStart(builder): builder.StartObject(2)
+ # BatchMatMulOptions
+ def AsymmetricQuantizeInputs(self):
+ o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
+ if o != 0:
+ return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
+ return False
+
+def BatchMatMulOptionsStart(builder): builder.StartObject(3)
def BatchMatMulOptionsAddAdjX(builder, adjX): builder.PrependBoolSlot(0, adjX, 0)
def BatchMatMulOptionsAddAdjY(builder, adjY): builder.PrependBoolSlot(1, adjY, 0)
+def BatchMatMulOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs): builder.PrependBoolSlot(2, asymmetricQuantizeInputs, 0)
def BatchMatMulOptionsEnd(builder): return builder.EndObject()