From 513784fcb52e1cbdd2197fba20c3d7183f06b336 Mon Sep 17 00:00:00 2001
From: Sannidhya Chauhan <sannidhya@google.com>
Date: Wed, 10 Jun 2026 05:39:30 -0700
Subject: [PATCH] Add enable_continuous_profiling option

PiperOrigin-RevId: 929790363
---
 tsl/profiler/lib/profiler_interface.h | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tsl/profiler/lib/profiler_interface.h b/tsl/profiler/lib/profiler_interface.h
index 2b0b71242..298dd91b5 100644
--- a/tsl/profiler/lib/profiler_interface.h
+++ b/tsl/profiler/lib/profiler_interface.h
@@ -15,12 +15,20 @@ limitations under the License.
 #ifndef TENSORFLOW_TSL_PROFILER_LIB_PROFILER_INTERFACE_H_
 #define TENSORFLOW_TSL_PROFILER_LIB_PROFILER_INTERFACE_H_
 
+#include <any>
+#include <cstddef>
+
 #include "xla/tsl/platform/status.h"
 #include "tsl/profiler/protobuf/xplane.pb.h"
 
 namespace tsl {
 namespace profiler {
 
+struct ConsumeResult {
+  std::any data;
+  size_t estimated_size_bytes = 0;
+};
+
 // Interface for tensorflow profiler plugins.
 //
 // ProfileSession calls each of these methods at most once per instance, and