| /external/tensorflow/tensorflow/python/eager/ |
| D | remote_cluster_test.py | 211 def worker_fn(i): function
215 worker_fn.get_concrete_function(x1)
219 y = worker_fn(x1)
225 y = worker_fn(x2)
233 def worker_fn(i): function
240 worker_fn.get_concrete_function(x1)
245 y = worker_fn(x1)
251 y = worker_fn(x1)
264 def worker_fn(i): function
268 worker_fn.get_concrete_function(x1)
[all …]
|
| D | remote_test.py | 549 def worker_fn(): function
555 self.assertAllEqual(worker_fn(), 9)
558 self.assertAllEqual(worker_fn(), 8)
620 def worker_fn(): function
626 self.assertAllEqual(worker_fn(), 9)
629 self.assertAllEqual(worker_fn(), 8)
646 def worker_fn(): function
652 self.assertAllEqual(worker_fn(), 9)
654 self.assertAllEqual(worker_fn(), 8)
660 worker_fn().numpy()
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/integration_test/ |
| D | mwms_peer_failure_test.py | 79 def worker_fn(): function
90 mpr = multi_process_runner.MultiProcessRunner(worker_fn, cluster_spec)
113 def worker_fn(): function
123 mpr = multi_process_runner.MultiProcessRunner(worker_fn, cluster_spec)
138 def worker_fn(attempts): function
152 worker_fn, cluster_spec, args=(attempts,), auto_restart=True)
160 def worker_fn(attempts): function
173 worker_fn, cluster_spec, args=(attempts,), auto_restart=True)
186 def worker_fn(attempts): function
212 worker_fn, cluster_spec, args=(attempts,), auto_restart=True)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator.py | 326 def _run_single_worker(worker_fn, argument
358 return worker_fn(strategy)
360 return worker_fn(strategy)
454 def _run_between_graph_client(worker_fn, strategy, eval_fn, eval_strategy, argument
476 args=(worker_fn, strategy, cluster_spec, task_type, task_id,
497 def _run_in_graph_client(worker_fn, strategy, eval_fn, eval_strategy, argument
514 worker_fn,
631 def run_distribute_coordinator(worker_fn, argument
788 _run_single_worker(worker_fn, strategy, None, None, None, session_config,
799 eval_fn = eval_fn or worker_fn
[all …]
|
| D | multi_worker_continuous_run_test.py | 92 def worker_fn(): function
101 worker_fn,
125 def worker_fn(): function
134 worker_fn,
|
| D | distribute_coordinator_test.py | 243 def _wrapped_worker_fn(self, worker_fn): argument
246 return worker_fn(*args, **kwargs)
249 def _run_coordinator_in_thread(self, worker_fn, strategy, **kwargs): argument
252 args=(self._wrapped_worker_fn(worker_fn), strategy),
257 def _run_multiple_coordinator_in_threads(self, worker_fn, strategy, argument
264 worker_fn,
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | cluster_coordinator_test.py | 486 def worker_fn(iterator): function
493 worker_fn, args=(iter(distributed_dataset),))
497 worker_fn, args=(iter(distributed_dataset),))
514 def worker_fn(iterator): function
527 self.coordinator.schedule(worker_fn, args=(iterator,))
534 self.coordinator.schedule(worker_fn, args=(iterator,))
558 def worker_fn(iterator): function
563 worker_fn, args=(iter(distributed_dataset),))
624 def worker_fn(var): function
630 self.coordinator.schedule(worker_fn, args=(worker_local_var,))
[all …]
|
| D | fault_tolerance_test.py | 452 def worker_fn(): function
455 remote_value = self.cluster_coord.schedule(worker_fn)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_multi_worker_test.py | 71 def worker_fn(): function
89 mpr = multi_process_runner.MultiProcessRunner(worker_fn, cluster_spec)
95 def worker_fn(): function
101 mpr = multi_process_runner.MultiProcessRunner(worker_fn, cluster_spec)
109 def worker_fn(): function
141 worker_fn, cluster_spec, auto_restart=True)
147 def worker_fn(): function
153 mpr = multi_process_runner.MultiProcessRunner(worker_fn, cluster_spec)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | parameter_server_training_test.py | 176 def worker_fn(iterator): function
199 self.coordinator.schedule(worker_fn, args=(distributed_iterator,))
|