From 4c4df2cf18955b2cc76f432b5f1321083d5bfb11 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Fri, 9 Dec 2022 04:11:05 -0800 Subject: [PATCH] Internal change for profiling PiperOrigin-RevId: 494135244 --- mediapipe/framework/profiler/graph_profiler.h | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mediapipe/framework/profiler/graph_profiler.h b/mediapipe/framework/profiler/graph_profiler.h index 29969af2..23caed4e 100644 --- a/mediapipe/framework/profiler/graph_profiler.h +++ b/mediapipe/framework/profiler/graph_profiler.h @@ -232,6 +232,11 @@ class GraphProfiler : public std::enable_shared_from_this { const ProfilerConfig& profiler_config() { return profiler_config_; } + // Helper method to expose the config to other profilers. + const ValidatedGraphConfig* GetValidatedGraphConfig() { + return validated_graph_; + } + private: // This can be used to add packet info for the input streams to the graph. // It treats the stream defined by |stream_name| as a stream produced by a