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
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,
}