aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite_mapping.py
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/vela/tflite_mapping.py
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/vela/tflite_mapping.py')
-rw-r--r--ethosu/vela/tflite_mapping.py21
1 files changed, 21 insertions, 0 deletions
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),
}