Searched refs:setup_tpu_session (Results 1 – 1 of 1) sorted by relevance
140 def setup_tpu_session(cluster_resolver): function2241 setup_tpu_session(strategy._tpu_cluster_resolver)