diff options
author | Henrik Hoglind <henrik.hoglind@arm.com> | 2021-11-23 11:55:48 +0100 |
---|---|---|
committer | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2021-11-23 11:46:50 +0000 |
commit | 3db0ffab211153c15c4e629789e9a2ab11f33506 (patch) | |
tree | 37be4f31fbff53a802da5d68a514b024730948a1 | |
parent | 482392198cc5852cd4757d1f83228fc64378312f (diff) | |
download | ethos-u-linux-driver-stack-3db0ffab211153c15c4e629789e9a2ab11f33506.tar.gz |
Add support for int32 in inference_runner21.11
Change-Id: Ic4605a22547ab797c638bfddd9719c69d495da75
-rw-r--r-- | driver_library/src/ethosu.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
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: |