diff options
author | Rickard Bolin <rickard.bolin@arm.com> | 2021-12-20 08:35:23 +0000 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2022-01-21 12:45:02 +0000 |
commit | 2de898a85d800f0e812cc13eff6363e27a377cf5 (patch) | |
tree | 4c4b81cf5c4c521d33af59234deffce00611ba78 /ethosu/vela/tflite_mapping.py | |
parent | 3c25ff658cf847ccfcb2d4d5796ffbe13a511894 (diff) | |
download | ethos-u-vela-2de898a85d800f0e812cc13eff6363e27a377cf5.tar.gz |
MLBEDSW-4870: Update to TensorFlow 2.7
Update the flatbuffers generated code to comply with TensorFlow 2.7
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: Iff29b05a6e145245861329b4ff9fc9fbd968da53
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index 396ed5ec..2a70b06c 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -31,6 +31,7 @@ from .tflite import AddNOptions from .tflite import AddOptions from .tflite import ArgMaxOptions from .tflite import ArgMinOptions +from .tflite import AssignVariableOptions from .tflite import BatchMatMulOptions from .tflite import BatchToSpaceNDOptions from .tflite import BidirectionalSequenceLSTMOptions @@ -96,8 +97,10 @@ from .tflite import PadV2Options from .tflite import Pool2DOptions from .tflite import PowOptions from .tflite import QuantizeOptions +from .tflite import RandomOptions from .tflite import RangeOptions from .tflite import RankOptions +from .tflite import ReadVariableOptions from .tflite import ReducerOptions from .tflite import ReshapeOptions from .tflite import ResizeBilinearOptions @@ -133,6 +136,7 @@ from .tflite import TransposeOptions from .tflite import UnidirectionalSequenceLSTMOptions from .tflite import UniqueOptions from .tflite import UnpackOptions +from .tflite import VarHandleOptions from .tflite import WhereOptions from .tflite import WhileOptions from .tflite import ZerosLikeOptions @@ -305,8 +309,13 @@ builtin_options_map = { BuiltinOptions.HashtableFindOptions: HashtableFindOptions.HashtableFindOptions, BuiltinOptions.HashtableImportOptions: HashtableImportOptions.HashtableImportOptions, BuiltinOptions.HashtableSizeOptions: HashtableSizeOptions.HashtableSizeOptions, + BuiltinOptions.VarHandleOptions: VarHandleOptions.VarHandleOptions, + BuiltinOptions.ReadVariableOptions: ReadVariableOptions.ReadVariableOptions, + BuiltinOptions.AssignVariableOptions: AssignVariableOptions.AssignVariableOptions, + BuiltinOptions.RandomOptions: RandomOptions.RandomOptions, } + builtin_options_inv_map = inverse_map(builtin_options_map) @@ -904,6 +913,19 @@ builtin_operator_map = { ), TFLITE_NO_INDICES, ), + BuiltinOperator.VAR_HANDLE: ( + Op.VarHandle, + OptionsSerializer("VarHandleOptions", ("container", "shared_name",),), + TFLITE_NO_INDICES, + ), + BuiltinOperator.READ_VARIABLE: (Op.ReadVariable, OptionsSerializer("ReadVariableOptions"), TFLITE_NO_INDICES), + BuiltinOperator.ASSIGN_VARIABLE: (Op.AssignVariable, OptionsSerializer("AssignVariableOptions"), TFLITE_NO_INDICES), + BuiltinOperator.BROADCAST_ARGS: (Op.BroadcastArgs, None, TFLITE_NO_INDICES), + BuiltinOperator.RANDOM_STANDARD_NORMAL: ( + Op.RandomStandardNormal, + OptionsSerializer("RandomOptions", ("seed", "seed2",),), + TFLITE_NO_INDICES, + ), BuiltinOperator.CUSTOM: (Op.Custom, CustomOptionsSerializer(), TFLITE_NO_INDICES), } |