Searched refs:initialize_tpu_system (Results 1 – 19 of 19) sorted by relevance
86 tpu_strategy_util.initialize_tpu_system(resolver)98 tpu_strategy_util.initialize_tpu_system(resolver)101 tpu_strategy_util.initialize_tpu_system(resolver)252 topology = tpu_strategy_util.initialize_tpu_system(resolver)363 topology = tpu_strategy_util.initialize_tpu_system(resolver)404 topology = tpu_strategy_util.initialize_tpu_system(resolver)1124 tpu_strategy_util.initialize_tpu_system(resolver)1252 topology = tpu_strategy_util.initialize_tpu_system(resolver)1321 topology = tpu_strategy_util.initialize_tpu_system(resolver)1357 topology = tpu_strategy_util.initialize_tpu_system(resolver)[all …]
48 tpu_strategy_util.initialize_tpu_system(resolver)
107 _topology = tpu_strategy_util.initialize_tpu_system(resolver)
1119 tpu_strategy_util.initialize_tpu_system(self._tpu_cluster_resolver)
955 tpu_strategy_util.initialize_tpu_system(resolver)
20 name: "initialize_tpu_system"
43 tpu_strategy_util.initialize_tpu_system(resolver)
82 tpu_strategy_util.initialize_tpu_system(resolver)
40 name: "initialize_tpu_system"
53 tpu_strategy_util.initialize_tpu_system(resolver)
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)311 tpu_strategy_util.initialize_tpu_system(self.resolver)512 tpu_strategy_util.initialize_tpu_system(self.resolver)
43 def initialize_tpu_system(cluster_resolver=None): function
131 tpu_strategy_util.initialize_tpu_system(self.resolver)
74 tpu_strategy_util.initialize_tpu_system(resolver)
123 tpu_strategy_util.initialize_tpu_system()
113 tpu_strategy_util.initialize_tpu_system(resolver)
70 tpu_strategy_util.initialize_tpu_system(resolver)
1029 disable_tfrt = True, # TODO(tfrt-devs): Update this test to call initialize_tpu_system
3371 …* TPUs can now be re-initialized multiple times, using `tf.tpu.experimental.initialize_tpu_system`.