cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
PiperOrigin-RevId: 575020084
This commit is contained in:
Sebastian Schmidt 2023-10-19 15:27:38 -07:00 committed by Copybara-Service
parent 1601073cf0
commit 305d7abec4
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mediapipe/tasks/c/vision/image_classifier/image_classifier.h
View File

@ -57,14 +57,17 @@ struct ImageFrame {
}; };
// TODO: Add GPU buffer declaration and proccessing logic for it. // 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. // The object to contain an image, realizes `OneOf` concept.
struct MpImage { struct MpImage {
enum { IMAGE_FRAME, GPU_BUFFER } type; enum { IMAGE_FRAME, GPU_BUFFER } type;
union { union {
ImageFrame image_frame; struct ImageFrame image_frame;
GpuBuffer gpu_buffer; struct GpuBuffer gpu_buffer;
}; };
}; };