From 86d9ef4e7aab28cc05725c0b6c3b954ba53f81c4 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 9 Jan 2026 11:04:25 -0800 Subject: [PATCH] Add GetUptime() for fetching the up time after init_main() is called. PiperOrigin-RevId: 854263428 --- tsl/platform/init_main.h | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tsl/platform/init_main.h b/tsl/platform/init_main.h index c02c1e8d1..58b5d4c59 100644 --- a/tsl/platform/init_main.h +++ b/tsl/platform/init_main.h @@ -16,11 +16,17 @@ limitations under the License. #ifndef TENSORFLOW_TSL_PLATFORM_INIT_MAIN_H_ #define TENSORFLOW_TSL_PLATFORM_INIT_MAIN_H_ +#include +#include namespace tsl { namespace port { void InitMain(const char* usage, int* argc, char*** argv); +const std::vector& GetArgvs(); + +const char* GetArgv0(); + } // namespace port } // namespace tsl