diff options
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index 79521680..55351cb9 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -143,7 +143,7 @@ datatype_map = { 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 @@ datatype_map_numpy = { TensorType.FLOAT16: np.float16, TensorType.FLOAT32: np.float32, TensorType.BOOL: np.bool, + TensorType.COMPLEX64: np.complex64, } |