From 031f4c1e5f78cef4e660e3a3ba5c8e401abbe2e8 Mon Sep 17 00:00:00 2001
From: Henrik Hoglind <henrik.hoglind@arm.com>
Date: Thu, 25 Mar 2021 08:32:23 +0100
Subject: Add support for float32 in inference_runner

Change-Id: I1c618be058863a3a2457a849e70b22851aa91b3a
---
 driver_library/src/ethosu.cpp | 2 ++
 1 file changed, 2 insertions(+)

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");
     }
-- 
cgit v1.2.1