From 305d7abec40aa70907270f4195df7f44d6924e53 Mon Sep 17 00:00:00 2001
From: Sebastian Schmidt <mrschmidt@google.com>
Date: Thu, 19 Oct 2023 15:27:38 -0700
Subject: [PATCH] Add a field to GPUBuffer C struct so FFIGen can handle it

PiperOrigin-RevId: 575020084
---
 .../tasks/c/vision/image_classifier/image_classifier.h   | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/mediapipe/tasks/c/vision/image_classifier/image_classifier.h b/mediapipe/tasks/c/vision/image_classifier/image_classifier.h
index 9b59e4127..60dc4a2c4 100644
--- a/mediapipe/tasks/c/vision/image_classifier/image_classifier.h
+++ b/mediapipe/tasks/c/vision/image_classifier/image_classifier.h
@@ -57,14 +57,17 @@ struct ImageFrame {
 };
 
 // TODO: Add GPU buffer declaration and proccessing logic for it.
-struct GpuBuffer {};
+struct GpuBuffer {
+  int width;
+  int height;
+};
 
 // The object to contain an image, realizes `OneOf` concept.
 struct MpImage {
   enum { IMAGE_FRAME, GPU_BUFFER } type;
   union {
-    ImageFrame image_frame;
-    GpuBuffer gpu_buffer;
+    struct ImageFrame image_frame;
+    struct GpuBuffer gpu_buffer;
   };
 };