Searched refs:_initialize_multi_worker (Results 1 – 3 of 3) sorted by relevance
102 self._initialize_multi_worker(cluster_resolver)150 def _initialize_multi_worker(self, cluster_resolver): member in CollectiveAllReduceExtended379 self._initialize_multi_worker(cluster_resolver)
111 self._initialize_multi_worker(cluster_resolver)115 def _initialize_multi_worker(self, cluster_resolver): member in ParameterServerStrategyExtended488 self._initialize_multi_worker(cluster_resolver)
461 self._initialize_multi_worker(devices)477 def _initialize_multi_worker(self, devices): member in MirroredExtended684 self._initialize_multi_worker(multi_worker_devices)