summaryrefslogtreecommitdiff
path: root/tools/tflitefile_tool
diff options
context:
space:
mode:
author오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>2018-10-25 09:09:52 +0900
committer박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>2018-10-25 09:09:52 +0900
commite8d0af54fcc31bd09be6a81d0534b23230246626 (patch)
treeb2592786b8da740c02ded5e0a34a24e8e7dab4a8 /tools/tflitefile_tool
parentd63df34263c946f6a1840f10c7920881ac4e2233 (diff)
downloadnnfw-e8d0af54fcc31bd09be6a81d0534b23230246626.tar.gz
nnfw-e8d0af54fcc31bd09be6a81d0534b23230246626.tar.bz2
nnfw-e8d0af54fcc31bd09be6a81d0534b23230246626.zip
Fix python syntax error (#3326)
Fix python syntax error in select_operator.py Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
Diffstat (limited to 'tools/tflitefile_tool')
-rwxr-xr-xtools/tflitefile_tool/select_operator.py42
1 files changed, 23 insertions, 19 deletions
diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py
index 0ad382377..7430388fe 100755
--- a/tools/tflitefile_tool/select_operator.py
+++ b/tools/tflitefile_tool/select_operator.py
@@ -350,26 +350,30 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
tflite.L2NormOptions.L2NormOptionsStart(new_builder)
tflite.L2NormOptions.L2NormOptionsAddFusedActivationFunction(
- new_builder, L2NormOptions.FusedActivationFunction())
+ new_builder, l2norm_option.FusedActivationFunction())
return tflite.L2NormOptions.L2NormOptionsEnd(new_builder)
# LocalResponseNormalizationOptions
- import tflite.LocalResponseNormalizationOptions
- if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().LocalResponseNormalizationOptions:
-
- lrn_option = tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptions()
- lrn_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
-
- tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsStart(new_builder)
- tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsAddRadius(
- new_builder, LocalResponseNormalizationOptions.Radius())
- tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsAddBias(
- new_builder, LocalResponseNormalizationOptions.Bias())
- tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsAddAlpha(
- new_builder, LocalResponseNormalizationOptions.Alpha())
- tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsAddBeta(
- new_builder, LocalResponseNormalizationOptions.Beta())
- return tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsEnd(new_builder)
+ import tflite.LocalResponseNormalizationOptions
+ if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions(
+ ).LocalResponseNormalizationOptions:
+
+ lrn_option = tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptions(
+ )
+ lrn_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+ tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsStart(
+ new_builder)
+ tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsAddRadius(
+ new_builder, lrn_option.Radius())
+ tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsAddBias(
+ new_builder, lrn_option.Bias())
+ tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsAddAlpha(
+ new_builder, lrn_option.Alpha())
+ tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsAddBeta(
+ new_builder, lrn_option.Beta())
+ return tflite.LocalResponseNormalizationOptions.LocalResponseNormalizationOptionsEnd(
+ new_builder)
# LSTMOptions: not supported
@@ -422,7 +426,7 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
tflite.SpaceToDepthOptions.SpaceToDepthOptionsStart(new_builder)
tflite.SpaceToDepthOptions.SpaceToDepthOptionsAddBlockSize(
- new_builder, SpaceToDepthOptions.BlockSize())
+ new_builder, space_to_depth_option.BlockSize())
return tflite.SpaceToDepthOptions.SpaceToDepthOptionsEnd(new_builder)
# EmbeddingLookupSparseOptions: not supported
@@ -584,7 +588,7 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
tflite.SplitOptions.SplitOptionsStart(new_builder)
tflite.SplitOptions.SplitOptionsAddNumSplits(new_builder,
- SplitOptions.NumSplits())
+ split_option.NumSplits())
return tflite.SplitOptions.SplitOptionsEnd(new_builder)
# LogSoftmaxOptions: not supported