From 3db0ffab211153c15c4e629789e9a2ab11f33506 Mon Sep 17 00:00:00 2001
From: Henrik Hoglind <henrik.hoglind@arm.com>
Date: Tue, 23 Nov 2021 11:55:48 +0100
Subject: Add support for int32 in inference_runner

Change-Id: Ic4605a22547ab797c638bfddd9719c69d495da75
---
 driver_library/src/ethosu.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/driver_library/src/ethosu.cpp b/driver_library/src/ethosu.cpp
index 4fc431b..29cb77e 100644
--- a/driver_library/src/ethosu.cpp
+++ b/driver_library/src/ethosu.cpp
@@ -116,6 +116,7 @@ size_t getTensorTypeSize(const enum tflite::TensorType type) {
         return 1;
     case tflite::TensorType::TensorType_INT16:
         return 2;
+    case tflite::TensorType::TensorType_INT32:
     case tflite::TensorType::TensorType_FLOAT32:
         return 4;
     default:
-- 
cgit v1.2.1