[model_parser] Prints optons for Depthwise_Conv_2D (#3613)
author윤현식/동작제어Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>
Mon, 19 Nov 2018 02:02:46 +0000 (11:02 +0900)
committer박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Mon, 19 Nov 2018 02:02:46 +0000 (11:02 +0900)
This commit adds feature to print optons for Depthwise_Conv_2D.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
tools/tflitefile_tool/option_printer.py

index 2ba0dac..08754f1 100644 (file)
@@ -54,3 +54,16 @@ class OptionPrinter(object):
                 "Dilation W:H = {}:{}".format(self.options.DilationWFactor(),
                                               self.options.DilationHFactor()),
                 "Padding = {}".format(self.GetPadding())))
+
+        elif (self.op_name == "DEPTHWISE_CONV_2D"):
+            print("{}Options".format(tab))
+
+            # yapf: disable
+            print("{}\t{}, {}, {}, {}".format(
+                tab, "Stride W:H = {}:{}".format(self.options.StrideW(),
+                                                 self.options.StrideH()),
+                "Dilation W:H = {}:{}".format(self.options.DilationWFactor(),
+                                              self.options.DilationHFactor()),
+                "Padding = {}".format(self.GetPadding()),
+                "DepthMultiplier = {}".format(self.options.DepthMultiplier())))
+            # yapf: enable