| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | tfconfig_cluster_resolver_test.py | 25 from tensorflow.python.distribute.cluster_resolver.tfconfig_cluster_resolver import TFConfigCluster…
62 cluster_resolver = TFConfigClusterResolver()
70 actual_cluster_spec = cluster_resolver.cluster_spec()
87 cluster_resolver = TFConfigClusterResolver()
88 self.assertEqual('ps0:2222', cluster_resolver.master())
104 cluster_resolver = TFConfigClusterResolver()
105 self.assertEqual('worker1:2222', cluster_resolver.master('worker', 1))
122 cluster_resolver = TFConfigClusterResolver()
123 self.assertEqual('sessionmaster:2222', cluster_resolver.master())
140 cluster_resolver = TFConfigClusterResolver()
[all …]
|
| D | sagemaker_cluster_resolver_test.py | 23 from tensorflow.python.distribute.cluster_resolver.sagemaker_cluster_resolver import SageMakerClust…
50 cluster_resolver = SageMakerClusterResolver()
55 actual_cluster_spec = cluster_resolver.cluster_spec()
62 cluster_resolver = SageMakerClusterResolver()
63 self.assertEqual('algo-1:2223', cluster_resolver.master())
69 cluster_resolver = SageMakerClusterResolver()
70 self.assertEqual('algo-2:2223', cluster_resolver.master('worker', 1))
76 cluster_resolver = SageMakerClusterResolver(rpc_layer='grpc')
77 self.assertEqual('grpc://algo-1:2223', cluster_resolver.master())
83 cluster_resolver = SageMakerClusterResolver(task_type='worker', task_id=0)
[all …]
|
| D | __init__.py | 28 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import ClusterResolver
29 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import SimpleClusterResolver
30 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import UnionClusterResolver
31 from tensorflow.python.distribute.cluster_resolver.gce_cluster_resolver import GCEClusterResolver
32 from tensorflow.python.distribute.cluster_resolver.kubernetes_cluster_resolver import KubernetesClu…
33 from tensorflow.python.distribute.cluster_resolver.slurm_cluster_resolver import SlurmClusterResolv…
34 from tensorflow.python.distribute.cluster_resolver.tfconfig_cluster_resolver import TFConfigCluster…
35 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| D | kubernetes_cluster_resolver_test.py | 21 from tensorflow.python.distribute.cluster_resolver.kubernetes_cluster_resolver import KubernetesClu…
78 cluster_resolver = KubernetesClusterResolver(
82 actual_cluster_spec = cluster_resolver.cluster_spec()
97 cluster_resolver = KubernetesClusterResolver(
101 actual_cluster_spec = cluster_resolver.cluster_spec()
118 cluster_resolver = KubernetesClusterResolver(
121 cluster_resolver.task_type = 'worker'
122 cluster_resolver.task_id = 0
123 self.assertEqual(cluster_resolver.task_type, 'worker')
124 self.assertEqual(cluster_resolver.task_id, 0)
[all …]
|
| D | kubernetes_cluster_resolver.py | 21 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import ClusterResolver
22 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import format_master_url
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/ |
| D | tpu_cluster_resolver_test.py | 28 from tensorflow.python.distribute.cluster_resolver.tpu import tpu_cluster_resolver as resolver
164 cluster_resolver = resolver.TPUClusterResolver(
172 actual_cluster_spec = cluster_resolver.cluster_spec()
182 """ % cluster_resolver._coordinator_port
184 self.assertEqual(cluster_resolver.master(), 'grpc://10.1.2.3:8470')
198 cluster_resolver = resolver.TPUClusterResolver(
206 actual_cluster_spec = cluster_resolver.cluster_spec()
211 self.assertEqual(cluster_resolver.master(), 'grpc://10.1.2.3:8470')
224 cluster_resolver = resolver.TPUClusterResolver(
233 cluster_resolver.cluster_spec()
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | utils.py | 28 def start_server(cluster_resolver, protocol): argument
32 if not (cluster_resolver.task_type == 'worker' or
33 cluster_resolver.task_type == 'ps'):
35 cluster_resolver.task_type))
38 cluster_resolver.cluster_spec().as_cluster_def(),
39 job_name=cluster_resolver.task_type,
40 task_index=cluster_resolver.task_id,
44 cluster_resolver.task_type, cluster_resolver.task_id)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_strategy_util.py | 23 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
41 def initialize_tpu_system(cluster_resolver=None): argument
56 if cluster_resolver is None:
64 cluster_resolver = TPUClusterResolver("")
65 assert isinstance(cluster_resolver, TPUClusterResolver)
67 tpu_name = compat.as_text(cluster_resolver._tpu) # pylint: disable=protected-access
84 job = "{}/replica:0/task:0".format(cluster_resolver.get_job_name())
116 master = cluster_resolver.master()
117 cluster_spec = cluster_resolver.cluster_spec()
143 def shutdown_tpu_system(cluster_resolver=None): argument
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_multi_worker_test.py | 29 from tensorflow.python.distribute import cluster_resolver as cluster_resolver_lib
42 def enable_collective_ops(cluster_resolver): argument
47 cluster=cluster_resolver.cluster_spec().as_cluster_def(),
49 job_name=cluster_resolver.task_type,
50 task_index=cluster_resolver.task_id,
51 protocol=cluster_resolver.rpc_layer or "grpc")
55 def enable_collective_ops_with_barrier(cluster_resolver): argument
57 enable_collective_ops(cluster_resolver)
110 cluster_resolver = cluster_resolver_lib.TFConfigClusterResolver()
111 enable_collective_ops(cluster_resolver)
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_v2_test.py | 31 from tensorflow.python.distribute.cluster_resolver import SimpleClusterResolver
52 cls.cluster_resolver = SimpleClusterResolver(ClusterSpec(cluster_def))
62 self.cluster_resolver)
96 self.cluster_resolver)
109 self.cluster_resolver)
115 self.cluster_resolver)
123 self.cluster_resolver)
151 cls.cluster_resolver = SimpleClusterResolver(ClusterSpec(cluster_def))
160 self.cluster_resolver)
168 self.cluster_resolver, sharded_variable.FixedShardsPartitioner(2))
[all …]
|
| D | collective_all_reduce_strategy.py | 41 from tensorflow.python.distribute.cluster_resolver import ClusterResolver
42 from tensorflow.python.distribute.cluster_resolver import SimpleClusterResolver
43 from tensorflow.python.distribute.cluster_resolver import TFConfigClusterResolver
168 cluster_resolver=None, argument
188 cluster_resolver=cluster_resolver,
207 def cluster_resolver(self): member in CollectiveAllReduceStrategy
240 cluster_resolver=None): argument
255 self).__init__(cluster_resolver, communication_options)
281 cluster_resolver=None): argument
288 cluster_resolver=cluster_resolver,
[all …]
|
| D | parameter_server_strategy.py | 34 from tensorflow.python.distribute.cluster_resolver import SimpleClusterResolver
35 from tensorflow.python.distribute.cluster_resolver import TFConfigClusterResolver
104 def __init__(self, cluster_resolver=None): argument
112 if cluster_resolver is None:
113 cluster_resolver = TFConfigClusterResolver()
116 self, cluster_resolver=cluster_resolver))
167 cluster_resolver=None, argument
172 cluster_resolver=cluster_resolver,
181 cluster_resolver=None, argument
184 if cluster_resolver and cluster_resolver.cluster_spec():
[all …]
|
| D | parameter_server_strategy_v2.py | 393 def __init__(self, cluster_resolver, variable_partitioner=None): argument
434 self._cluster_resolver = cluster_resolver
435 self._extended = ParameterServerStrategyV2Extended(self, cluster_resolver,
437 self._verify_args_and_config(cluster_resolver)
441 "with cluster_spec: %s", cluster_resolver.cluster_spec())
485 def _verify_args_and_config(self, cluster_resolver): argument
486 if not cluster_resolver.cluster_spec():
492 cluster_spec = cluster_resolver.cluster_spec()
497 cluster_resolver.task_type,
498 cluster_resolver.task_id)
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.UnionResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.UnionCluster…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.SimpleClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.SimpleCluste…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.KubernetesClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.kubernetes_cluster_resolver.K…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.TFConfigClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.tfconfig_cluster_resolver.TFC…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.GCEClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.gce_cluster_resolver.GCEClust…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.SlurmClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.slurm_cluster_resolver.SlurmC…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.UnionResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.UnionCluster…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.SimpleClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.SimpleCluste…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.KubernetesClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.kubernetes_cluster_resolver.K…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.GCEClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.gce_cluster_resolver.GCEClust…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.TFConfigClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.tfconfig_cluster_resolver.TFC…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|
| D | tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.SlurmClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.slurm_cluster_resolver.SlurmC…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|