From 1b64b55a92360a51673bc28b3167d58f1c5795c2 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9C=A4=ED=98=84=EC=8B=9D/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Principal=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Mon, 19 Nov 2018 11:02:46 +0900 Subject: [PATCH] [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 --- tools/tflitefile_tool/option_printer.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) 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 -- 2.7.4