commit f767b937c12935be3cb1f9ee406fbb796176a40c
author Jacob Bohlin <jacob.bohlin@arm.com> Thu Aug 13 15:32:45 2020 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Tue Aug 18 13:51:46 2020 +0000
tree 3d27a0d63b582f2a1c4d23866a2823ce40d6f1ff
parent 0b8268a0dac80aa22133ca83ed6912d3b565439a

MLBEDSW-2732: Added complex64 to datatypes

Added complex64 datatype to allow pass through without crashing.

Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I8beeceafb32182d4877a9880d21d51ba21033030

ethosu/vela/data_type.py

diff --git a/ethosu/vela/data_type.py b/ethosu/vela/data_type.py
index bb4c558..4d05fef 100644
--- a/ethosu/vela/data_type.py
+++ b/ethosu/vela/data_type.py
@@ -35,6 +35,7 @@
     String = 128
     Resource = 256
     Variant = 512
+    Complex = 1024
 
 
 class DataType:
@@ -78,6 +79,7 @@
         BaseType.String: ("string", False),
         BaseType.Resource: ("resource", False),
         BaseType.Variant: ("variant", False),
+        BaseType.Complex: ("complex%s", True),
     }
 
 
@@ -112,3 +114,4 @@
 DataType.bool = DataType(BaseType.Bool, 8)
 DataType.resource = DataType(BaseType.Resource, 8)
 DataType.variant = DataType(BaseType.Variant, 8)
+DataType.complex64 = DataType(BaseType.Complex, 64)

ethosu/vela/tflite_mapping.py

diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index 7952168..55351cb 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -143,7 +143,7 @@
     TensorType.FLOAT32: DataType.float32,
     TensorType.STRING: DataType.string,
     TensorType.BOOL: DataType.bool,
-    # no TensorType.COMPLEX64 for now
+    TensorType.COMPLEX64: DataType.complex64,
 }
 
 datatype_inv_map = inverse_map(datatype_map)
@@ -163,6 +163,7 @@
     TensorType.FLOAT16: np.float16,
     TensorType.FLOAT32: np.float32,
     TensorType.BOOL: np.bool,
+    TensorType.COMPLEX64: np.complex64,
 }