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