From: Shubham Gupta/System SW /SRI-Bangalore/Engineer/삼성전자 <shub98.gupta@samsung.com>
Date: Wed, 24 Oct 2018 01:08:34 +0000 (+0530)
Subject: [tfltool] Op selector for LRN (#3284)
X-Git-Tag: 0.3~544
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=08041253a939faa55c9a9055bff01f5307053ae6;p=platform%2Fcore%2Fml%2Fnnfw.git

[tfltool] Op selector for LRN (#3284)

This patch will provide support to select LRN op in select_operator.py

Signed-off-by: shubham <shub98.gupta@samsung.com>
---

diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py
index fb9267f..0ad3823 100755
--- a/tools/tflitefile_tool/select_operator.py
+++ b/tools/tflitefile_tool/select_operator.py
@@ -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