experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Initialize the TPU library when loading through tpu_executor_dlsym_initializer

Browse Source 
PiperOrigin-RevId: 346133750
Change-Id: Ia0108df839c3d0fc11aaaea453286e3d228828d1
This commit is contained in:
Frank Chen 2020-12-07 11:08:04 -08:00 committed by TensorFlower Gardener
parent b0e7a83e21
commit e9deb8b804
1 changed files with 1 additions and 1 deletions

2
tensorflow/core/tpu/tpu_executor_dlsym_initializer.cc
View File

@ -47,7 +47,7 @@ Status InitializeTpuLibrary(void* library_handle) {
void (*initialize_fn)(bool init_library, int argc, char** argv);
initialize_fn = reinterpret_cast<decltype(initialize_fn)>(
dlsym(library_handle, "TfTpu_Initialize"));
(*initialize_fn)(/*init_library=*/false, /*argc=*/0, /*argv=*/nullptr);
(*initialize_fn)(/*init_library=*/true, /*argc=*/0, /*argv=*/nullptr);
RegisterTpuPlatform();
}