Searched refs:initialize_tpu_system (Results 1 – 17 of 17) sorted by relevance
83 tpu_strategy_util.initialize_tpu_system(resolver)95 tpu_strategy_util.initialize_tpu_system(resolver)98 tpu_strategy_util.initialize_tpu_system(resolver)199 topology = tpu_strategy_util.initialize_tpu_system(resolver)310 topology = tpu_strategy_util.initialize_tpu_system(resolver)351 topology = tpu_strategy_util.initialize_tpu_system(resolver)975 tpu_strategy_util.initialize_tpu_system(resolver)1103 topology = tpu_strategy_util.initialize_tpu_system(resolver)1172 topology = tpu_strategy_util.initialize_tpu_system(resolver)1208 topology = tpu_strategy_util.initialize_tpu_system(resolver)[all …]
48 tpu_strategy_util.initialize_tpu_system(resolver)
102 _topology = tpu_strategy_util.initialize_tpu_system(resolver)
1118 tpu_strategy_util.initialize_tpu_system(self._tpu_cluster_resolver)
913 tpu_strategy_util.initialize_tpu_system(resolver)
20 name: "initialize_tpu_system"
82 tpu_strategy_util.initialize_tpu_system(resolver)
40 name: "initialize_tpu_system"
41 def initialize_tpu_system(cluster_resolver=None): function
77 tpu_strategy_util.initialize_tpu_system(self.resolver)188 tpu_strategy_util.initialize_tpu_system(self.resolver)198 tpu_strategy_util.initialize_tpu_system(self.resolver)310 tpu_strategy_util.initialize_tpu_system(self.resolver)509 tpu_strategy_util.initialize_tpu_system(self.resolver)
131 tpu_strategy_util.initialize_tpu_system(self.resolver)
75 tpu_strategy_util.initialize_tpu_system(resolver)
113 tpu_strategy_util.initialize_tpu_system(resolver)
53 tf.tpu.experimental.initialize_tpu_system(resolver)
74 tpu_strategy_util.initialize_tpu_system(resolver)
1596 …* TPUs can now be re-initialized multiple times, using `tf.tpu.experimental.initialize_tpu_system`.