Searched refs:_initialize_single_worker (Results 1 – 2 of 2) sorted by relevance
383 super(CollectiveAllReduceExtended, self)._initialize_single_worker(498 super(CollectiveAllReduceExtended, self)._initialize_single_worker(
354 self._initialize_single_worker(devices)382 def _initialize_single_worker(self, devices): member in MirroredExtended