Add uint16 support to TfLiteTypeSizeOf (#2693)

BUG=uint16 missing in TfLiteTypeSizeOf
tensorflow · Sep 23, 2024 · ce26927 · ce26927
1 parent 6147f20
commit ce26927
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tensorflow/lite/micro/memory_helpers.cc b/tensorflow/lite/micro/memory_helpers.cc
@@ -1,4 +1,4 @@
-/* Copyright 2023 The TensorFlow Authors. All Rights Reserved.
+/* Copyright 2024 The TensorFlow Authors. All Rights Reserved.
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
@@ -71,6 +71,9 @@ TfLiteStatus TfLiteTypeSizeOf(TfLiteType type, size_t* size) {
     case kTfLiteUInt8:
       *size = sizeof(uint8_t);
       break;
+    case kTfLiteUInt16:
+      *size = sizeof(uint16_t);
+      break;
     case kTfLiteInt8:
       *size = sizeof(int8_t);
       break;