From: 윤현식/동작제어Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>
Date: Mon, 19 Nov 2018 02:02:46 +0000 (+0900)
Subject: [model_parser] Prints optons for Depthwise_Conv_2D (#3613)
X-Git-Tag: 0.3~396
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=1b64b55a92360a51673bc28b3167d58f1c5795c2;p=platform%2Fcore%2Fml%2Fnnfw.git

[model_parser] Prints optons for Depthwise_Conv_2D (#3613)

This commit adds feature to print optons for Depthwise_Conv_2D.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
---

diff --git a/tools/tflitefile_tool/option_printer.py b/tools/tflitefile_tool/option_printer.py
index 2ba0dac..08754f1 100644
--- a/tools/tflitefile_tool/option_printer.py
+++ b/tools/tflitefile_tool/option_printer.py
@@ -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