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

Restore TpuPlatform auto registration code.

Browse Source 
PiperOrigin-RevId: 317409587
Change-Id: If44d7a39a45c4c7026f70a4d79d965a54c4db295
This commit is contained in:
Wenhao Jia 2020-06-19 17:27:50 -07:00 committed by TensorFlower Gardener
parent f840a62268
commit 715b02167d
1 changed files with 8 additions and 0 deletions

8
tensorflow/stream_executor/tpu/tpu_platform.cc
View File

@ -134,4 +134,12 @@ void RegisterTpuPlatform() {
}
}
REGISTER_MODULE_INITIALIZER(tpu_platform, RegisterTpuPlatform());
// Note that module initialization sequencing is not supported in the
// open-source project, so this will be a no-op there.
REGISTER_MODULE_INITIALIZER_SEQUENCE(tpu_platform, multi_platform_manager);
REGISTER_MODULE_INITIALIZER_SEQUENCE(multi_platform_manager_listener,
tpu_platform);
} // namespace tensorflow