[tool] Enable select operator for reduce operators (#3810)
author오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Mon, 3 Dec 2018 11:28:48 +0000 (20:28 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Mon, 3 Dec 2018 11:28:48 +0000 (20:28 +0900)
Enable select operator for reduce operators: reduce_max, reduce_min, reduce_any, mean, sum,

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
tools/tflitefile_tool/select_operator.py

index 12deac1..b39aaf7 100755 (executable)
@@ -502,7 +502,18 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
         return tflite.SpaceToBatchNDOptions.SpaceToBatchNDOptionsEnd(new_builder)
 
     # TransposeOptions: not supported
-    # ReducerOptions: not supported
+
+    # ReducerOptions
+    import tflite.ReducerOptions
+    if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().ReducerOptions:
+
+        reducer_option = tflite.ReducerOptions.ReducerOptions()
+        reducer_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+        tflite.ReducerOptions.ReducerOptionsStart(new_builder)
+        tflite.ReducerOptions.ReducerOptionsAddKeepDims(new_builder,
+                                                        reducer_option.KeepDims())
+        return tflite.ReducerOptions.ReducerOptionsEnd(new_builder)
 
     # SubOptions
     import tflite.SubOptions