| commit | 3db0ffab211153c15c4e629789e9a2ab11f33506 | [log] [tgz] |
|---|---|---|
| author | Henrik Hoglind <henrik.hoglind@arm.com> | Tue Nov 23 11:55:48 2021 +0100 |
| committer | Kristofer Jonsson <kristofer.jonsson@arm.com> | Tue Nov 23 11:46:50 2021 +0000 |
| tree | 37be4f31fbff53a802da5d68a514b024730948a1 | |
| parent | 482392198cc5852cd4757d1f83228fc64378312f [diff] |
Add support for int32 in inference_runner Change-Id: Ic4605a22547ab797c638bfddd9719c69d495da75
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 @@ return 1; case tflite::TensorType::TensorType_INT16: return 2; + case tflite::TensorType::TensorType_INT32: case tflite::TensorType::TensorType_FLOAT32: return 4; default: