diff options
author | William Isaksson <william.isaksson@arm.com> | 2023-09-11 13:00:30 +0000 |
---|---|---|
committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-09-13 10:51:49 +0000 |
commit | f0cb1abcab287a282a238d660089e23bd0c93c23 (patch) | |
tree | 52e2ac758a38855624eeee528f095f5c444be9f0 /ethosu/vela/tflite_mapping.py | |
parent | c0bb868fe375ff38eede8be363165794ca780978 (diff) | |
download | ethos-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.py | 21 |
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), } |