[tflite2tf] add TRANSPOSE

Author: peterjc123

contrib/tflite2tf/README.md

@@ -15,6 +15,7 @@ A utility for model conversion between TFLite and Tensorflow SavedModel.
 - TRANSPOSE_CONV
 - SLICE
 - RELU
+- TRANSPOSE
 
 ### Usage
 ```py

contrib/tflite2tf/tflite2tf.py

@@ -56,6 +56,14 @@ def parse_reshape(op: tflite.Operator, transform_tensors: typing.Set[int]):
     return line
 
 
+def parse_transpose(op: tflite.Operator, transform_tensors: typing.Set[int]):
+    line = (
+        f'{output_name(op, 0, transform_tensors)} = tf.transpose({input_name(op, 0, transform_tensors)},'
+        f' {input_name(op, 1, transform_tensors)})'
+    )
+    return line
+
+
 def parse_resize(op: tflite.Operator, transform_tensors: typing.Set[int]):
     line = (
         f'{output_name(op, 0, transform_tensors)} = tf.image.resize({input_name(op, 0, transform_tensors)},'
@@ -214,6 +222,7 @@ def parse_slice(op: tflite.Operator, transform_tensors: typing.Set[int]):
     'TRANSPOSE_CONV': parse_transposeconv2d,
     'SLICE': parse_slice,
     'RELU': parse_relu,
+    'TRANSPOSE': parse_transpose,
 }
 
 # Header for the generated script