Home
last modified time | relevance | path

Searched defs:worker_fn (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dremote_cluster_test.py211 def worker_fn(i): function
233 def worker_fn(i): function
264 def worker_fn(i): function
292 def worker_fn(i): function
325 def worker_fn(i): function
341 def worker_fn(i): function
371 def worker_fn(i): function
415 def worker_fn(i): function
453 def worker_fn(i): function
473 def worker_fn(i): function
[all …]
Dremote_test.py550 def worker_fn(): function
621 def worker_fn(): function
664 def worker_fn(): function
703 def worker_fn(): function
/external/tensorflow/tensorflow/python/distribute/integration_test/
Dmwms_peer_failure_test.py82 def worker_fn(): function
118 def worker_fn(): function
145 def worker_fn(attempts): function
172 def worker_fn(attempts): function
203 def worker_fn(attempts): function
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator_test.py511 def worker_fn(iterator): function
541 def worker_fn(iterator): function
586 def worker_fn(iterator): function
667 def worker_fn(var): function
712 def worker_fn(iterator): function
728 def worker_fn(iterator): function
753 def worker_fn(iterator1, iterator2): function
776 def worker_fn(iterator1, iterator2): function
992 def worker_fn(iterator): function
1045 def worker_fn(): function
[all …]
Dfault_tolerance_test.py511 def worker_fn(): function
527 def worker_fn(): function
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_multi_worker_test.py71 def worker_fn(): function
95 def worker_fn(): function
109 def worker_fn(): function
147 def worker_fn(): function
/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_continuous_run_test.py92 def worker_fn(): function
125 def worker_fn(): function
Ddistribute_coordinator.py326 def _run_single_worker(worker_fn, argument
454 def _run_between_graph_client(worker_fn, strategy, eval_fn, eval_strategy, argument
497 def _run_in_graph_client(worker_fn, strategy, eval_fn, eval_strategy, argument
631 def run_distribute_coordinator(worker_fn, argument
Ddistribute_coordinator_test.py243 def _wrapped_worker_fn(self, worker_fn): argument
249 def _run_coordinator_in_thread(self, worker_fn, strategy, **kwargs): argument
257 def _run_multiple_coordinator_in_threads(self, worker_fn, strategy, argument
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistribute_coordinator_utils.py291 def _run_single_worker(worker_fn, argument
448 def run_distribute_coordinator(worker_fn, argument