[tfltool] Update option parsing in select operator for v1.12 (#3594)
author오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Thu, 15 Nov 2018 07:10:18 +0000 (16:10 +0900)
committer박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Thu, 15 Nov 2018 07:10:18 +0000 (16:10 +0900)
Add comment for new options
Fix DepthwiseConv2DOptions for additional field

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

index 7430388..27030e9 100755 (executable)
@@ -264,6 +264,10 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
             new_builder, depthconv2d_option.DepthMultiplier())
         tflite.DepthwiseConv2DOptions.DepthwiseConv2DOptionsAddFusedActivationFunction(
             new_builder, depthconv2d_option.FusedActivationFunction())
+        tflite.DepthwiseConv2DOptions.DepthwiseConv2DOptionsAddDilationWFactor(
+            new_builder, depthconv2d_option.DilationWFactor())
+        tflite.DepthwiseConv2DOptions.DepthwiseConv2DOptionsAddDilationHFactor(
+            new_builder, depthconv2d_option.DilationHFactor())
         return tflite.DepthwiseConv2DOptions.DepthwiseConv2DOptionsEnd(new_builder)
 
     # ConcatEmbeddingsOptions: not supported
@@ -660,6 +664,16 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
     # PowOptions: not supported
     # ArgMinOptions: not supported
     # FakeQuantOptions: not supported
+    # PackOptions: not supported
+    # LogicalOrOptions: not supported
+    # OneHotOptions: not supported
+    # LogicalAndOptions: not supported
+    # LogicalNotOptions: not supported
+    # UnpackOptions: not supported
+    # FloorDivOptions: not supported
+    # SquareOptions: not supported
+    # ZerosLikeOptions: not supported
+    # FillOptions: not supported
 
     # Cannot handle builtin option type yet
     print("Cannot handle this option yet")