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 <string>
+#include <vector>
 namespace tsl {
 namespace port {
 
 void InitMain(const char* usage, int* argc, char*** argv);
 
+const std::vector<std::string>& GetArgvs();
+
+const char* GetArgv0();
+
 }  // namespace port
 }  // namespace tsl