| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | multi_worker_util.py | 22 def normalize_cluster_spec(cluster_spec): argument
45 def task_count(cluster_spec, task_type): argument
52 def _validate_cluster_spec(cluster_spec, argument
108 def is_chief(cluster_spec=None, task_type=None, task_id=None): argument
148 def collective_leader(cluster_spec, task_type, task_id): argument
182 def coordination_leader(cluster_spec): argument
207 def worker_count(cluster_spec, task_type): argument
228 def id_in_cluster(cluster_spec, task_type, task_id): argument
|
| H A D | distribute_coordinator.py | 91 def _get_num_workers(cluster_spec): argument
110 cluster_spec, argument
277 def cluster_spec(self): member in _WorkerContext
324 cluster_spec, argument
359 def _split_cluster_for_evaluator(cluster_spec, task_type): argument
381 def _run_std_server(cluster_spec=None, argument
451 cluster_spec, session_config, rpc_layer): argument
494 cluster_spec, session_config, rpc_layer): argument
526 strategy, eval_strategy, session_config, cluster_spec, task_type, task_id): argument
632 cluster_spec=None, argument
|
| H A D | distribute_coordinator_test.py | 99 cluster_spec=None, argument
244 cluster_spec, **kwargs): argument
397 cluster_spec=None, argument
763 def _thread_fn(cluster_spec): argument
787 def _run_mock_server(cluster_spec=None, argument
903 def _mock_run_std_server(cluster_spec=None, argument
|
| H A D | multi_process_runner.py | 125 cluster_spec, argument
271 cluster_spec=None, argument
420 cluster_spec=None, argument
897 def __init__(self, cluster_spec, initializer=None, share_gpu=True): argument
1141 def _set_tf_config(task_type, task_id, cluster_spec, rpc_layer=None): argument
1157 cluster_spec, argument
|
| H A D | multi_worker_test_base.py | 525 def _run_between_graph_clients(self, client_fn, cluster_spec, num_gpus, *args, argument
672 def _run_task_in_thread(self, task_fn, cluster_spec, task_type, task_id, argument
714 def run_multiple_tasks_in_threads(self, task_fn, cluster_spec, *args, argument
734 def _run_task_in_process(self, cmd_args, cluster_spec, task_type, task_id): argument
750 def run_multiple_tasks_in_processes(self, cmd_args, cluster_spec): argument
|
| H A D | estimator_training.py | 36 def _count_ps(cluster_spec): argument
45 def _count_worker(cluster_spec, chief_task_type): argument
55 def _get_global_id(cluster_spec, task_type, task_id, chief_task_type): argument
|
| H A D | mirrored_strategy.py | 94 def _cluster_spec_to_device_list(cluster_spec, num_gpus_per_worker): argument
681 cluster_spec=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | distribute_coordinator_utils.py | 56 def _get_num_workers(cluster_spec): argument
75 cluster_spec, argument
242 def cluster_spec(self): member in _WorkerContext
289 cluster_spec, argument
324 def _split_cluster_for_evaluator(cluster_spec, task_type): argument
345 def _run_std_server(cluster_spec=None, argument
415 session_config, cluster_spec, argument
448 cluster_spec=None, argument
655 def normalize_cluster_spec(cluster_spec): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| H A D | cluster_resolver.py | 95 def cluster_spec(self): member in ClusterResolver
320 def __init__(self, cluster_spec, master='', task_type=None, task_id=None, argument
341 def cluster_spec(self): member in SimpleClusterResolver
495 def cluster_spec(self): member in UnionClusterResolver
|
| H A D | cluster_resolver_test.py | 32 def cluster_spec(self): member in MockBaseClusterResolver
123 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| H A D | kubernetes_cluster_resolver.py | 140 def cluster_spec(self): member in KubernetesClusterResolver
|
| H A D | sagemaker_cluster_resolver_test.py | 30 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| H A D | slurm_cluster_resolver_test.py | 48 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| H A D | kubernetes_cluster_resolver_test.py | 49 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| H A D | tfconfig_cluster_resolver_test.py | 33 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| H A D | gce_cluster_resolver.py | 125 def cluster_spec(self): member in GCEClusterResolver
|
| H A D | gce_cluster_resolver_test.py | 28 def _verifyClusterSpecEquality(self, cluster_spec, expected_proto): argument
|
| H A D | slurm_cluster_resolver.py | 300 def cluster_spec(self): member in SlurmClusterResolver
|
| H A D | tfconfig_cluster_resolver.py | 149 def cluster_spec(self): member in TFConfigClusterResolver
|
| H A D | sagemaker_cluster_resolver.py | 156 def cluster_spec(self): member in SageMakerClusterResolver
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/ |
| H A D | grpc_tensorflow_server.py | 42 def parse_cluster_spec(cluster_spec, cluster, verbose=False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/failure_handling/ |
| H A D | failure_handler_test.py | 118 cluster_spec, argument
|
| H A D | gce_failure_handler_test.py | 106 cluster_spec, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/ |
| H A D | tpu_cluster_resolver.py | 304 def cluster_spec(self): member in TPUClusterResolver
|