tree 898722c1028e0131d458c42418f443ca95db2ea5
parent 2861f8595abcae680ab6d1db88e41783a4750be7
author Fiona Lang <flang@google.com> 1685984223 -0700
committer Copybara-Service <copybara-worker@google.com> 1685984790 -0700

Break the interdependency between tpu/tpu_strategy_util.py,  distribute/cluster_resolver/tpu/tpu_cluster_resolver.py, and eager/remote.py.

Make exported wrappers for initialize_tpu_system and shutdown_tpu_system in tpu_cluster_resolver.py, and make implementation functions for them in tpu_strategy_util.py that they pass a reference of TPUClusterResolver to. Add a test for this.

Replace an isinstance check of TPUClusterResolver in remote.py with an isinstance check of ClusterResolver and a hasattr check.

Make all inline imports regular imports and add all missing BUILD deps. Update references to the new location of initialize_tpu_system.

PiperOrigin-RevId: 537905952
GitOrigin-RevId: 6088a22ddb8fc299d18a34596b91ccf077b86b17
Change-Id: I06951d37f6f70a34df92adf7d80e3765078f2ce0