diff options
| author | Henrik Hoglind <henrik.hoglind@arm.com> | 2021-03-25 08:32:23 +0100 |
|---|---|---|
| committer | Henrik Hoglind <henrik.hoglind@arm.com> | 2021-03-25 08:32:23 +0100 |
| commit | 031f4c1e5f78cef4e660e3a3ba5c8e401abbe2e8 (patch) | |
| tree | a32afdc2945191e30b3e14506f1ecede5cf7d20a | |
| parent | 097d2929c1cbc872d1780d98f3f58bf2647fe51c (diff) | |
| download | ethos-u-linux-driver-stack-031f4c1e5f78cef4e660e3a3ba5c8e401abbe2e8.tar.gz | |
Add support for float32 in inference_runner
Change-Id: I1c618be058863a3a2457a849e70b22851aa91b3a
| -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"); } |