[model_parser] Print activation function as name (#2092)

Print Activation Function Type as its name, not the integer code.

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>

diff --git a/tools/tflitefile_tool/operator_wrapping.py b/tools/tflitefile_tool/operator_wrapping.py
index 118436a..780fa20 100755 (executable)
--- a/tools/tflitefile_tool/operator_wrapping.py
+++ b/tools/tflitefile_tool/operator_wrapping.py
@@ -3,6 +3,7 @@
 import tflite.Operator
 import tflite.OperatorCode
 import tflite.BuiltinOperator
+import tflite.ActivationFunctionType
 from tensor_wrapping import Tensor
 from operation import Operation
 from perf_predictor import PerfPredictor
@@ -24,6 +25,8 @@ def BuildEnumClassStrMap(obj):
 
 class EnumStrMaps():
     BuiltinOpcode = BuildEnumClassStrMap(tflite.BuiltinOperator.BuiltinOperator())
+    ActivationFunctionType = BuildEnumClassStrMap(
+        tflite.ActivationFunctionType.ActivationFunctionType())
 
 
 def GetStrTensorIndex(tensors):
@@ -124,8 +127,7 @@ class Operator(object):
         # fused activation function
         try:
             activation_code = options.FusedActivationFunction()
-            fused_activation = str(
-                activation_code)  # TODO print the name, not the integer code
+            fused_activation = EnumStrMaps.ActivationFunctionType[activation_code]
             print("\tFused Activation: " + fused_activation)
         except AttributeError:
             # This operator does not support FusedActivationFunction