diff options
Diffstat (limited to 'driver_library/src')
-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: |