Home
last modified time | relevance | path

Searched defs:cluster_spec (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_util.py25 def normalize_cluster_spec(cluster_spec): argument
49 def _validate_cluster_spec(cluster_spec, task_type, task_id): argument
75 def is_chief(cluster_spec, task_type, task_id): argument
104 def worker_count(cluster_spec, task_type): argument
125 def id_in_cluster(cluster_spec, task_type, task_id): argument
Ddistribute_coordinator.py95 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
382 def _run_std_server(cluster_spec=None, argument
452 cluster_spec, session_config, rpc_layer): argument
495 cluster_spec, session_config, rpc_layer): argument
527 strategy, eval_strategy, session_config, cluster_spec, task_type, task_id): argument
633 cluster_spec=None, argument
Destimator_training.py40 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
Ddistribute_coordinator_test.py107 cluster_spec=None, argument
248 cluster_spec, **kwargs): argument
392 cluster_spec=None, argument
769 def _thread_fn(cluster_spec): argument
793 def _run_mock_server(cluster_spec=None, argument
909 def _mock_run_std_server(cluster_spec=None, argument
Dcollective_all_reduce_strategy.py355 cluster_spec=None, argument
Dmirrored_strategy.py316 def _cluster_spec_to_device_list(cluster_spec, num_gpus_per_worker): argument
669 cluster_spec=None, argument
Dparameter_server_strategy.py461 cluster_spec=None, argument
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drun_config.py169 def cluster_spec(self): member in ClusterConfig
411 def _count_ps(cluster_spec): argument
416 def _count_worker(cluster_spec): argument
434 def _get_master(cluster_spec, task_type, task_id): argument
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_tensorflow_server.cc41 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/
Dcluster_resolver.py92 def cluster_spec(self): member in ClusterResolver
185 def __init__(self, cluster_spec, master='', task_type=None, task_id=None, argument
206 def cluster_spec(self): member in SimpleClusterResolver
329 def cluster_spec(self): member in UnionClusterResolver
Dcluster_resolver_test.py35 def cluster_spec(self): member in MockBaseClusterResolver
126 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
Dkubernetes_cluster_resolver.py122 def cluster_spec(self): member in KubernetesClusterResolver
Dslurm_cluster_resolver_test.py35 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
Dkubernetes_cluster_resolver_test.py53 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
Dslurm_cluster_resolver.py129 def cluster_spec(self): member in SlurmClusterResolver
Dtfconfig_cluster_resolver_test.py36 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
Dgce_cluster_resolver.py106 def cluster_spec(self): member in GCEClusterResolver
Dgce_cluster_resolver_test.py32 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
Dtfconfig_cluster_resolver.py124 def cluster_spec(self): member in TFConfigClusterResolver
Dtpu_cluster_resolver.py386 def cluster_spec(self): member in TPUClusterResolver
Dtpu_cluster_resolver_test.py96 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
/external/tensorflow/tensorflow/python/debug/lib/
Dgrpc_tensorflow_server.py45 def parse_cluster_spec(cluster_spec, cluster, verbose=False): argument
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmulti_worker_test_base.py289 def _run_between_graph_clients(self, client_fn, cluster_spec, num_gpus, *args, argument
397 def _run_task_in_thread(self, task_fn, cluster_spec, task_type, task_id, argument
418 def run_multiple_tasks_in_threads(self, task_fn, cluster_spec, *args, argument
Dkeras_multi_worker_test.py253 def _run_standalone_client(test_obj, strategy, cluster_spec): argument
Dcollective_all_reduce_strategy_test.py71 def create_test_objects(cluster_spec=None, argument

12