aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite_mapping.py
diff options
context:
space:
mode:
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 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),
}