Searched refs:_test_device_assignment_distributed (Results 1 – 1 of 1) sorted by relevance
123 def _test_device_assignment_distributed(self, task_type, task_id, num_gpus): member in ParameterServerStrategyTestBase608 self._test_device_assignment_distributed('worker', 1, num_gpus)