diff --git a/tensorflow/core/tpu/libtftpu.h b/tensorflow/core/tpu/libtftpu.h
index 9171af87061..72868571e72 100644
--- a/tensorflow/core/tpu/libtftpu.h
+++ b/tensorflow/core/tpu/libtftpu.h
@@ -41,7 +41,8 @@ limitations under the License.
 extern "C" {
 #endif
 
-TFTPU_CAPI_EXPORT void TfTpu_Initialize(bool init_library);
+TFTPU_CAPI_EXPORT void TfTpu_Initialize(bool init_library, int argc,
+                                        char** argv);
 
 #ifdef __cplusplus
 }
diff --git a/tensorflow/core/tpu/tpu_api_dlsym_initializer.cc b/tensorflow/core/tpu/tpu_api_dlsym_initializer.cc
index e4d723305a9..f67d885b4d1 100644
--- a/tensorflow/core/tpu/tpu_api_dlsym_initializer.cc
+++ b/tensorflow/core/tpu/tpu_api_dlsym_initializer.cc
@@ -52,7 +52,7 @@ Status InitializeTpuLibrary(void* library_handle) {
     void (*initialize_fn)(bool init_library);
     initialize_fn = reinterpret_cast<decltype(initialize_fn)>(
         dlsym(library_handle, "TfTpu_Initialize"));
-    (*initialize_fn)(/*init_library=*/true);
+    (*initialize_fn)(/*init_library=*/true, /*argc=*/0, /*argv=*/nullptr);
 
     RegisterTpuPlatform();
     RegisterTpuSystemDevice();
diff --git a/tensorflow/core/tpu/tpu_executor_dlsym_initializer.cc b/tensorflow/core/tpu/tpu_executor_dlsym_initializer.cc
index 4d84781f4e3..ecdd66edf03 100644
--- a/tensorflow/core/tpu/tpu_executor_dlsym_initializer.cc
+++ b/tensorflow/core/tpu/tpu_executor_dlsym_initializer.cc
@@ -45,8 +45,8 @@ Status InitializeTpuLibrary(void* library_handle) {
   // supporting library providing the necessary APIs.
   if (s.ok()) {
     void (*initialize_fn)();
-    initialize_fn = reinterpret_cast<decltype(initialize_fn)>(
-        dlsym(library_handle, "TfTpu_Initialize"));
+    initialize_fn = reinterpret_cast<decltype(initialize_fn)>(dlsym(
+        library_handle, "TfTpu_Initialize", /*argc=*/0, /*argv=*/nullptr));
     (*initialize_fn)();
 
     RegisterTpuPlatform();