| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_util.py | 26 def normalize_cluster_spec(cluster_spec): argument
49 def task_count(cluster_spec, task_type): argument
56 def _validate_cluster_spec(cluster_spec, argument
112 def is_chief(cluster_spec=None, task_type=None, task_id=None): argument
152 def collective_leader(cluster_spec, task_type, task_id): argument
186 def worker_count(cluster_spec, task_type): argument
207 def id_in_cluster(cluster_spec, task_type, task_id): argument
|
| D | distribute_coordinator.py | 95 def _get_num_workers(cluster_spec): argument
114 cluster_spec, argument
281 def cluster_spec(self): member in _WorkerContext
328 cluster_spec, argument
363 def _split_cluster_for_evaluator(cluster_spec, task_type): argument
385 def _run_std_server(cluster_spec=None, argument
455 cluster_spec, session_config, rpc_layer): argument
498 cluster_spec, session_config, rpc_layer): argument
530 strategy, eval_strategy, session_config, cluster_spec, task_type, task_id): argument
636 cluster_spec=None, argument
|
| D | estimator_training.py | 40 def _count_ps(cluster_spec): argument
49 def _count_worker(cluster_spec, chief_task_type): argument
59 def _get_global_id(cluster_spec, task_type, task_id, chief_task_type): argument
|
| D | distribute_coordinator_test.py | 110 cluster_spec=None, argument
258 cluster_spec, **kwargs): argument
411 cluster_spec=None, argument
777 def _thread_fn(cluster_spec): argument
801 def _run_mock_server(cluster_spec=None, argument
917 def _mock_run_std_server(cluster_spec=None, argument
|
| D | multi_process_runner.py | 127 cluster_spec, argument
267 cluster_spec=None, argument
406 cluster_spec=None, argument
879 def __init__(self, cluster_spec, initializer=None): argument
1108 def _set_tf_config(task_type, task_id, cluster_spec, rpc_layer=None): argument
1124 cluster_spec, argument
|
| D | multi_worker_test_base.py | 547 def _run_between_graph_clients(self, client_fn, cluster_spec, num_gpus, *args, argument
694 def _run_task_in_thread(self, task_fn, cluster_spec, task_type, task_id, argument
736 def run_multiple_tasks_in_threads(self, task_fn, cluster_spec, *args, argument
756 def _run_task_in_process(self, cmd_args, cluster_spec, task_type, task_id): argument
772 def run_multiple_tasks_in_processes(self, cmd_args, cluster_spec): argument
|
| D | mirrored_strategy.py | 86 def _cluster_spec_to_device_list(cluster_spec, num_gpus_per_worker): argument
675 cluster_spec=None, argument
|
| D | collective_all_reduce_strategy_test.py | 69 def create_test_objects(cluster_spec=None, argument
|
| D | collective_all_reduce_strategy.py | 657 cluster_spec=None, argument
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensorflow_server.cc | 41 Status FillServerDef(const string& cluster_spec, const string& job_name, in FillServerDef()
99 tensorflow::string cluster_spec; in main() local
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver.py | 99 def cluster_spec(self): member in ClusterResolver
324 def __init__(self, cluster_spec, master='', task_type=None, task_id=None, argument
345 def cluster_spec(self): member in SimpleClusterResolver
499 def cluster_spec(self): member in UnionClusterResolver
|
| D | cluster_resolver_test.py | 36 def cluster_spec(self): member in MockBaseClusterResolver
127 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| D | kubernetes_cluster_resolver.py | 149 def cluster_spec(self): member in KubernetesClusterResolver
|
| D | sagemaker_cluster_resolver_test.py | 34 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| D | slurm_cluster_resolver_test.py | 52 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| D | kubernetes_cluster_resolver_test.py | 53 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| D | tfconfig_cluster_resolver_test.py | 37 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| D | gce_cluster_resolver.py | 129 def cluster_spec(self): member in GCEClusterResolver
|
| D | gce_cluster_resolver_test.py | 32 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| D | slurm_cluster_resolver.py | 304 def cluster_spec(self): member in SlurmClusterResolver
|
| D | tfconfig_cluster_resolver.py | 153 def cluster_spec(self): member in TFConfigClusterResolver
|
| D | sagemaker_cluster_resolver.py | 160 def cluster_spec(self): member in SageMakerClusterResolver
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | grpc_tensorflow_server.py | 45 def parse_cluster_spec(cluster_spec, cluster, verbose=False): argument
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/ |
| D | tpu_cluster_resolver.py | 305 def cluster_spec(self): member in TPUClusterResolver
|
| D | tpu_cluster_resolver_test.py | 103 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|