diff options
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 dba30c4..647430e 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), } |