Searched refs:worker_fn (Results 1 – 3 of 3) sorted by relevance
| /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)
451 def _run_between_graph_client(worker_fn, strategy, eval_fn, eval_strategy, argument
473 args=(worker_fn, strategy, cluster_spec, task_type, task_id,
494 def _run_in_graph_client(worker_fn, strategy, eval_fn, eval_strategy, argument
511 worker_fn,
628 def run_distribute_coordinator(worker_fn, argument
778 _run_single_worker(worker_fn, strategy, None, None, None, session_config,
789 eval_fn = eval_fn or worker_fn
[all …]
|
| D | distribute_coordinator_test.py | 239 def _run_coordinator_in_thread(self, worker_fn, strategy, **kwargs): argument
242 args=(worker_fn, strategy),
247 def _run_multiple_coordinator_in_threads(self, worker_fn, strategy, argument
254 worker_fn,
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | keras_multi_worker_test.py | 258 def worker_fn(strategy): function
283 worker_fn,
|