add TRANSPOSE to convert_map

apache · Aug 4, 2019 · ccf91cc · ccf91cc
1 parent c63ad37
commit ccf91cc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py
@@ -81,7 +81,8 @@ def __init__(self, model, subgraph, exp_tab):
             'PAD': self.convert_pad,
             'PACK': self.convert_pack,
             'LOGISTIC': self.convert_logistic,
-            'SPLIT': self.convert_split
+            'SPLIT': self.convert_split,
+            'TRANSPOSE': self.convert_transpose
         }
 
     def check_unsupported_ops(self):