From b0a0251ba621d94ee2cd304a265b565e7c8f997e Mon Sep 17 00:00:00 2001 From: Sannidhya Chauhan Date: Wed, 20 May 2026 21:58:45 -0700 Subject: [PATCH] Implement Consume and Serialize methods in TPU tracer and counter collectors. PiperOrigin-RevId: 918823040 --- 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 +#include + #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