aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite_mapping.py
diff options
context:
space:
mode:
authorJonas Ohlsson <jonas.ohlsson@arm.com>2022-03-30 10:30:25 +0200
committerJonas Ohlsson <jonas.ohlsson@arm.com>2022-03-30 15:54:14 +0200
commitd85750702229af97c0b0bbda6e397a23254b6144 (patch)
tree389962105a35d5cef595cfeb5d640bd59a0d0ff8 /ethosu/vela/tflite_mapping.py
parentcc5f4de1c35ba44fca7ff6295c6ae846f8242344 (diff)
downloadethos-u-vela-d85750702229af97c0b0bbda6e397a23254b6144.tar.gz
Update version of Black to 22.3.0
Update version of Black to 22.3.0 due to updated dependencies. Updates to fix reported issues due to new version. Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com> Change-Id: I60056aae452093ce8dcea1f499ecced22b25eef1
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
-rw-r--r--ethosu/vela/tflite_mapping.py21
1 files changed, 18 insertions, 3 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index 2a70b06c..a5f7fa2b 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -669,7 +669,10 @@ builtin_operator_map = {
BuiltinOperator.DIV: (Op.Div, OptionsSerializer("DivOptions", (fused_act,)), TFLITE_NO_INDICES),
BuiltinOperator.SQUEEZE: (
Op.Squeeze,
- OptionsSerializer("SqueezeOptions", (("squeeze_dims", is_int_vec),),),
+ OptionsSerializer(
+ "SqueezeOptions",
+ (("squeeze_dims", is_int_vec),),
+ ),
TFLITE_IFM_INDICES,
),
BuiltinOperator.UNIDIRECTIONAL_SEQUENCE_LSTM: (
@@ -915,7 +918,13 @@ builtin_operator_map = {
),
BuiltinOperator.VAR_HANDLE: (
Op.VarHandle,
- OptionsSerializer("VarHandleOptions", ("container", "shared_name",),),
+ OptionsSerializer(
+ "VarHandleOptions",
+ (
+ "container",
+ "shared_name",
+ ),
+ ),
TFLITE_NO_INDICES,
),
BuiltinOperator.READ_VARIABLE: (Op.ReadVariable, OptionsSerializer("ReadVariableOptions"), TFLITE_NO_INDICES),
@@ -923,7 +932,13 @@ builtin_operator_map = {
BuiltinOperator.BROADCAST_ARGS: (Op.BroadcastArgs, None, TFLITE_NO_INDICES),
BuiltinOperator.RANDOM_STANDARD_NORMAL: (
Op.RandomStandardNormal,
- OptionsSerializer("RandomOptions", ("seed", "seed2",),),
+ OptionsSerializer(
+ "RandomOptions",
+ (
+ "seed",
+ "seed2",
+ ),
+ ),
TFLITE_NO_INDICES,
),
BuiltinOperator.CUSTOM: (Op.Custom, CustomOptionsSerializer(), TFLITE_NO_INDICES),