diff --git a/src/gallium/targets/teflon/test_executor.cpp b/src/gallium/targets/teflon/test_executor.cpp
index 4683d7a671e..f0035694f4f 100644
--- a/src/gallium/targets/teflon/test_executor.cpp
+++ b/src/gallium/targets/teflon/test_executor.cpp
@@ -526,6 +526,11 @@ run_model(TfLiteModel *model, enum executor executor, void ***input, size_t *num
          (*output_sizes)[i] = output_tensor->bytes / 4;
          break;
       }
+      case kTfLiteInt16:
+      case kTfLiteUInt16: {
+         (*output_sizes)[i] = output_tensor->bytes / 2;
+         break;
+      }
       default: {
          (*output_sizes)[i] = output_tensor->bytes;
          break;