From b53acf626759998ec3463f42dcb089c164f4b5f3 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Tue, 31 Jan 2023 11:18:30 -0800 Subject: [PATCH] Internal change PiperOrigin-RevId: 506059384 --- mediapipe/framework/calculator_graph.h | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mediapipe/framework/calculator_graph.h b/mediapipe/framework/calculator_graph.h index 8d58ff312..04f9de45f 100644 --- a/mediapipe/framework/calculator_graph.h +++ b/mediapipe/framework/calculator_graph.h @@ -53,10 +53,14 @@ #include "mediapipe/framework/port/status.h" #include "mediapipe/framework/scheduler.h" #include "mediapipe/framework/thread_pool_executor.pb.h" -#include "mediapipe/gpu/gpu_service.h" namespace mediapipe { +#if !MEDIAPIPE_DISABLE_GPU +class GpuResources; +struct GpuSharedData; +#endif // !MEDIAPIPE_DISABLE_GPU + typedef absl::StatusOr StatusOrPoller; // The class representing a DAG of calculator nodes.