Add a field to GPUBuffer C struct so FFIGen can handle it

PiperOrigin-RevId: 575020084
kp-forks · Oct 19, 2023 · 305d7ab · 305d7ab
1 parent 1601073
commit 305d7ab
Showing 1 changed file with 6 additions and 3 deletions.
diff --git 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;
   };
 };