| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | remote_cluster_test.py | 226 def worker_fn(i): function
248 def worker_fn(i): function
279 def worker_fn(i): function
307 def worker_fn(i): function
340 def worker_fn(i): function
356 def worker_fn(i): function
386 def worker_fn(i): function
430 def worker_fn(i): function
468 def worker_fn(i): function
488 def worker_fn(i): function
[all …]
|
| H A D | remote_test.py | 589 def worker_fn(): function
660 def worker_fn(): function
703 def worker_fn(): function
742 def worker_fn(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/ |
| H A D | cluster_coordinator_test.py | 547 def worker_fn(iterator): function
577 def worker_fn(iterator): function
622 def worker_fn(iterator): function
701 def worker_fn(var): function
746 def worker_fn(iterator): function
762 def worker_fn(iterator): function
787 def worker_fn(iterator1, iterator2): function
810 def worker_fn(iterator1, iterator2): function
1014 def worker_fn(iterator): function
1066 def worker_fn(iterator): function
[all …]
|
| H A D | fault_tolerance_test.py | 510 def worker_fn(): function
526 def worker_fn(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/integration_test/ |
| H A D | mwms_peer_failure_test.py | 76 def worker_fn(): function
111 def worker_fn(): function
137 def worker_fn(attempts): function
163 def worker_fn(attempts): function
193 def worker_fn(attempts): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | collective_ops_multi_worker_test.py | 67 def worker_fn(): function
91 def worker_fn(): function
105 def worker_fn(): function
143 def worker_fn(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | distributed_table_test.py | 174 def worker_fn(iterator): function
217 def worker_fn(iterator): function
258 def worker_fn(iterator): function
302 def worker_fn(iterator): function
|
| H A D | distribute_coordinator.py | 322 def _run_single_worker(worker_fn, argument
450 def _run_between_graph_client(worker_fn, strategy, eval_fn, eval_strategy, argument
493 def _run_in_graph_client(worker_fn, strategy, eval_fn, eval_strategy, argument
627 def run_distribute_coordinator(worker_fn, argument
|
| H A D | multi_worker_continuous_run_test.py | 88 def worker_fn(): function
121 def worker_fn(): function
|
| H A D | distribute_coordinator_test.py | 229 def _wrapped_worker_fn(self, worker_fn): argument
235 def _run_coordinator_in_thread(self, worker_fn, strategy, **kwargs): argument
243 def _run_multiple_coordinator_in_threads(self, worker_fn, strategy, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | distribute_coordinator_utils.py | 287 def _run_single_worker(worker_fn, argument
444 def run_distribute_coordinator(worker_fn, argument
|
| /aosp_15_r20/external/pytorch/test/distributed/rpc/ |
| H A D | test_share_memory.py | 58 def worker_fn(m): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/failure_handling/ |
| H A D | failure_handler_test.py | 116 def worker_fn(self, member in PreemptionCheckpointTest
|
| H A D | gce_failure_handler_test.py | 103 def worker_fn( member in GceFailureHandlingTest
|