diff --git a/tensorflow/stream_executor/tpu/tpu_platform.cc b/tensorflow/stream_executor/tpu/tpu_platform.cc index 13a845829c1..4bccd822e91 100644 --- a/tensorflow/stream_executor/tpu/tpu_platform.cc +++ b/tensorflow/stream_executor/tpu/tpu_platform.cc @@ -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