diff options
Diffstat (limited to 'tools/tflitefile_tool/option_printer.py')
| -rw-r--r-- | tools/tflitefile_tool/option_printer.py | 34 |
1 files changed, 16 insertions, 18 deletions
diff --git a/tools/tflitefile_tool/option_printer.py b/tools/tflitefile_tool/option_printer.py index 08754f1ce..15265adf2 100644 --- a/tools/tflitefile_tool/option_printer.py +++ b/tools/tflitefile_tool/option_printer.py @@ -35,35 +35,33 @@ class OptionPrinter(object): if (self.options == 0): return - if (self.op_name == "AVERAGE_POOL_2D" or self.op_name == "MAX_POOL_2D"): + option_str = self.GetOptionString() + if option_str: print("{}Options".format(tab)) + print("{}\t{}".format(tab, option_str)) - print("{}\t{}, {}, {}".format( - tab, "Filter W:H = {}:{}".format(self.options.FilterWidth(), - self.options.FilterHeight()), + def GetOptionString(self): + if (self.op_name == "AVERAGE_POOL_2D" or self.op_name == "MAX_POOL_2D"): + return "{}, {}, {}".format( + "Filter W:H = {}:{}".format(self.options.FilterWidth(), + self.options.FilterHeight()), "Stride W:H = {}:{}".format(self.options.StrideW(), self.options.StrideH()), - "Padding = {}".format(self.GetPadding()))) - + "Padding = {}".format(self.GetPadding())) elif (self.op_name == "CONV_2D"): - print("{}Options".format(tab)) - - print("{}\t{}, {}, {}".format( - tab, "Stride W:H = {}:{}".format(self.options.StrideW(), - self.options.StrideH()), + return "{}, {}, {}".format( + "Stride W:H = {}:{}".format(self.options.StrideW(), + self.options.StrideH()), "Dilation W:H = {}:{}".format(self.options.DilationWFactor(), self.options.DilationHFactor()), - "Padding = {}".format(self.GetPadding()))) - + "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(), + return "{}, {}, {}, {}".format( + "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()))) + "DepthMultiplier = {}".format(self.options.DepthMultiplier())) # yapf: enable |