diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py index 9ebc6ba325762..ef003b529d96e 100644 --- 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):