aboutsummaryrefslogtreecommitdiff
path: root/ethosu
diff options
context:
space:
mode:
authorWilliam Isaksson <william.isaksson@arm.com>2023-09-11 13:00:30 +0000
committerRickard Bolin <rickard.bolin@arm.com>2023-09-13 10:51:49 +0000
commitf0cb1abcab287a282a238d660089e23bd0c93c23 (patch)
tree52e2ac758a38855624eeee528f095f5c444be9f0 /ethosu
parentc0bb868fe375ff38eede8be363165794ca780978 (diff)
downloadethos-u-vela-f0cb1abcab287a282a238d660089e23bd0c93c23.tar.gz
MLBEDSW-8035: Update to TensorFlow 2.13
- Adds 3 ops: Bitcast, BitcastXor, RightShift Change-Id: Ia9721c69d4f3da0deba7526addb95a9a54e63adf Signed-off-by: William Isaksson <william.isaksson@arm.com>
Diffstat (limited to 'ethosu')
-rw-r--r--ethosu/vela/operation.py3
-rw-r--r--ethosu/vela/tflite/BitcastOptions.py36
-rw-r--r--ethosu/vela/tflite/BitwiseXorOptions.py36
-rw-r--r--ethosu/vela/tflite/BuiltinOperator.py3
-rw-r--r--ethosu/vela/tflite/BuiltinOptions.py3
-rw-r--r--ethosu/vela/tflite/RightShiftOptions.py36
-rw-r--r--ethosu/vela/tflite_mapping.py21
7 files changed, 138 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 31839c72..94d256c2 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -334,6 +334,9 @@ class Op(Enum):
UnsortedSegmentMax = OperatorInfo()
UnsortedSegmentMin = OperatorInfo()
UnsortedSegmentSum = OperatorInfo()
+ Bitcast = OperatorInfo()
+ BitwiseXor = OperatorInfo()
+ RightShift = OperatorInfo()
@property
def info(self):
diff --git a/ethosu/vela/tflite/BitcastOptions.py b/ethosu/vela/tflite/BitcastOptions.py
new file mode 100644
index 00000000..093dad7b
--- /dev/null
+++ b/ethosu/vela/tflite/BitcastOptions.py
@@ -0,0 +1,36 @@
+# automatically generated by the FlatBuffers compiler, do not modify
+
+# namespace: tflite
+
+import flatbuffers
+from flatbuffers.compat import import_numpy
+np = import_numpy()
+
+class BitcastOptions(object):
+ __slots__ = ['_tab']
+
+ @classmethod
+ def GetRootAs(cls, buf, offset=0):
+ n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
+ x = BitcastOptions()
+ x.Init(buf, n + offset)
+ return x
+
+ @classmethod
+ def GetRootAsBitcastOptions(cls, buf, offset=0):
+ """This method is deprecated. Please switch to GetRootAs."""
+ return cls.GetRootAs(buf, offset)
+ @classmethod
+ def BitcastOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+ return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
+
+ # BitcastOptions
+ def Init(self, buf, pos):
+ self._tab = flatbuffers.table.Table(buf, pos)
+
+def BitcastOptionsStart(builder): builder.StartObject(0)
+def Start(builder):
+ return BitcastOptionsStart(builder)
+def BitcastOptionsEnd(builder): return builder.EndObject()
+def End(builder):
+ return BitcastOptionsEnd(builder) \ No newline at end of file
diff --git a/ethosu/vela/tflite/BitwiseXorOptions.py b/ethosu/vela/tflite/BitwiseXorOptions.py
new file mode 100644
index 00000000..4b53d108
--- /dev/null
+++ b/ethosu/vela/tflite/BitwiseXorOptions.py
@@ -0,0 +1,36 @@
+# automatically generated by the FlatBuffers compiler, do not modify
+
+# namespace: tflite
+
+import flatbuffers
+from flatbuffers.compat import import_numpy
+np = import_numpy()
+
+class BitwiseXorOptions(object):
+ __slots__ = ['_tab']
+
+ @classmethod
+ def GetRootAs(cls, buf, offset=0):
+ n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
+ x = BitwiseXorOptions()
+ x.Init(buf, n + offset)
+ return x
+
+ @classmethod
+ def GetRootAsBitwiseXorOptions(cls, buf, offset=0):
+ """This method is deprecated. Please switch to GetRootAs."""
+ return cls.GetRootAs(buf, offset)
+ @classmethod
+ def BitwiseXorOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+ return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
+
+ # BitwiseXorOptions
+ def Init(self, buf, pos):
+ self._tab = flatbuffers.table.Table(buf, pos)
+
+def BitwiseXorOptionsStart(builder): builder.StartObject(0)
+def Start(builder):
+ return BitwiseXorOptionsStart(builder)
+def BitwiseXorOptionsEnd(builder): return builder.EndObject()
+def End(builder):
+ return BitwiseXorOptionsEnd(builder) \ No newline at end of file
diff --git a/ethosu/vela/tflite/BuiltinOperator.py b/ethosu/vela/tflite/BuiltinOperator.py
index 0cce0792..ec2c1472 100644
--- a/ethosu/vela/tflite/BuiltinOperator.py
+++ b/ethosu/vela/tflite/BuiltinOperator.py
@@ -162,3 +162,6 @@ class BuiltinOperator(object):
ATAN2 = 156
UNSORTED_SEGMENT_MIN = 157
SIGN = 158
+ BITCAST = 159
+ BITWISE_XOR = 160
+ RIGHT_SHIFT = 161
diff --git a/ethosu/vela/tflite/BuiltinOptions.py b/ethosu/vela/tflite/BuiltinOptions.py
index 06396b08..7c3b3da3 100644
--- a/ethosu/vela/tflite/BuiltinOptions.py
+++ b/ethosu/vela/tflite/BuiltinOptions.py
@@ -127,3 +127,6 @@ class BuiltinOptions(object):
UnsortedSegmentSumOptions = 121
ATan2Options = 122
SignOptions = 123
+ BitcastOptions = 124
+ BitwiseXorOptions = 125
+ RightShiftOptions = 126
diff --git a/ethosu/vela/tflite/RightShiftOptions.py b/ethosu/vela/tflite/RightShiftOptions.py
new file mode 100644
index 00000000..f347a604
--- /dev/null
+++ b/ethosu/vela/tflite/RightShiftOptions.py
@@ -0,0 +1,36 @@
+# automatically generated by the FlatBuffers compiler, do not modify
+
+# namespace: tflite
+
+import flatbuffers
+from flatbuffers.compat import import_numpy
+np = import_numpy()
+
+class RightShiftOptions(object):
+ __slots__ = ['_tab']
+
+ @classmethod
+ def GetRootAs(cls, buf, offset=0):
+ n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
+ x = RightShiftOptions()
+ x.Init(buf, n + offset)
+ return x
+
+ @classmethod
+ def GetRootAsRightShiftOptions(cls, buf, offset=0):
+ """This method is deprecated. Please switch to GetRootAs."""
+ return cls.GetRootAs(buf, offset)
+ @classmethod
+ def RightShiftOptionsBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+ return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x54\x46\x4C\x33", size_prefixed=size_prefixed)
+
+ # RightShiftOptions
+ def Init(self, buf, pos):
+ self._tab = flatbuffers.table.Table(buf, pos)
+
+def RightShiftOptionsStart(builder): builder.StartObject(0)
+def Start(builder):
+ return RightShiftOptionsStart(builder)
+def RightShiftOptionsEnd(builder): return builder.EndObject()
+def End(builder):
+ return RightShiftOptionsEnd(builder) \ No newline at end of file
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index dba30c46..647430ef 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -38,6 +38,8 @@ from .tflite import BatchMatMulOptions
from .tflite import BatchToSpaceNDOptions
from .tflite import BidirectionalSequenceLSTMOptions
from .tflite import BidirectionalSequenceRNNOptions
+from .tflite import BitcastOptions
+from .tflite import BitwiseXorOptions
from .tflite import BroadcastToOptions
from .tflite import BucketizeOptions
from .tflite import CallOnceOptions
@@ -113,6 +115,7 @@ from .tflite import ResizeNearestNeighborOptions
from .tflite import ReverseSequenceOptions
from .tflite import ReverseV2Options
from .tflite import Rfft2dOptions
+from .tflite import RightShiftOptions
from .tflite import RNNOptions
from .tflite import ScatterNdOptions
from .tflite import SegmentSumOptions
@@ -336,6 +339,9 @@ builtin_options_map = {
BuiltinOptions.UnsortedSegmentSumOptions: UnsortedSegmentSumOptions.UnsortedSegmentSumOptions,
BuiltinOptions.ATan2Options: ATan2Options.ATan2Options,
BuiltinOptions.SignOptions: SignOptions.SignOptions,
+ BuiltinOptions.BitcastOptions: BitcastOptions.BitcastOptions,
+ BuiltinOptions.BitwiseXorOptions: BitwiseXorOptions.BitwiseXorOptions,
+ BuiltinOptions.RightShiftOptions: RightShiftOptions.RightShiftOptions,
}
@@ -1022,6 +1028,21 @@ builtin_operator_map = {
OptionsSerializer("UnsortedSegmentSumOptions"),
TFLITE_NO_INDICES,
),
+ BuiltinOperator.BITCAST: (
+ Op.Bitcast,
+ OptionsSerializer("BitcastOptions"),
+ TFLITE_NO_INDICES,
+ ),
+ BuiltinOperator.BITWISE_XOR: (
+ Op.BitwiseXor,
+ OptionsSerializer("BitwiseXorOptions"),
+ TFLITE_NO_INDICES,
+ ),
+ BuiltinOperator.RIGHT_SHIFT: (
+ Op.RightShift,
+ OptionsSerializer("RightShiftOptions"),
+ TFLITE_NO_INDICES,
+ ),
BuiltinOperator.CUSTOM: (Op.Custom, CustomOptionsSerializer(), TFLITE_NO_INDICES),
}