diff options
Diffstat (limited to 'driver_library/src/ethosu.cpp')
| -rw-r--r-- | driver_library/src/ethosu.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/driver_library/src/ethosu.cpp b/driver_library/src/ethosu.cpp index 2c498a8..40fc909 100644 --- a/driver_library/src/ethosu.cpp +++ b/driver_library/src/ethosu.cpp @@ -64,6 +64,8 @@ size_t getTensorTypeSize(const enum tflite::TensorType type) { return 1; case tflite::TensorType::TensorType_INT16: return 2; + case tflite::TensorType::TensorType_FLOAT32: + return 4; default: throw EthosU::Exception("Unsupported tensor type"); } |