Searched defs:cluster_resolver (Results 1 – 7 of 7) sorted by relevance
37 def get_first_tpu_host_device(cluster_resolver): argument57 def initialize_tpu_system(cluster_resolver=None): argument
89 cluster_resolver=TFConfigClusterResolver()): argument100 def _initialize_strategy(self, cluster_resolver): argument106 def _initialize_local(self, cluster_resolver): argument150 def _initialize_multi_worker(self, cluster_resolver): argument
101 cluster_resolver=TFConfigClusterResolver()): argument109 def _initialize_strategy(self, cluster_resolver): argument115 def _initialize_multi_worker(self, cluster_resolver): argument206 def _initialize_local(self, cluster_resolver): argument
75 def get_workers_list(cluster_resolver): argument
63 def __init__(self, cluster_resolver): argument
73 def __init__(self, cluster_resolver): argument
140 def setup_tpu_session(cluster_resolver): argument