[tfltool] Op selector for LRN (#3284)
authorShubham Gupta/System SW /SRI-Bangalore/Engineer/삼성전자 <shub98.gupta@samsung.com>
Wed, 24 Oct 2018 01:08:34 +0000 (06:38 +0530)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Wed, 24 Oct 2018 01:08:34 +0000 (10:08 +0900)
This patch will provide support to select LRN op in select_operator.py

Signed-off-by: shubham <shub98.gupta@samsung.com>
tools/tflitefile_tool/select_operator.py

index fb9267f..0ad3823 100755 (executable)
@@ -353,7 +353,24 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
             new_builder, L2NormOptions.FusedActivationFunction())
         return tflite.L2NormOptions.L2NormOptionsEnd(new_builder)
 
-    # LocalResponseNormalizationOptions: not supported
+    # 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)
+
     # LSTMOptions: not supported
 
     # ResizeBilinearOptions