diff options
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 |
commit | e8d0af54fcc31bd09be6a81d0534b23230246626 (patch) | |
tree | b2592786b8da740c02ded5e0a34a24e8e7dab4a8 /tools/tflitefile_tool | |
parent | d63df34263c946f6a1840f10c7920881ac4e2233 (diff) | |
download | nnfw-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-x | tools/tflitefile_tool/select_operator.py | 42 |
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 |