| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | tfconfig_cluster_resolver_test.py | 25 from tensorflow.python.distribute.cluster_resolver import TFConfigClusterResolver
61 cluster_resolver = TFConfigClusterResolver()
69 actual_cluster_spec = cluster_resolver.cluster_spec()
86 cluster_resolver = TFConfigClusterResolver()
87 self.assertEqual('ps0:2222', cluster_resolver.master())
103 cluster_resolver = TFConfigClusterResolver()
104 self.assertEqual('worker1:2222', cluster_resolver.master('worker', 1))
121 cluster_resolver = TFConfigClusterResolver()
122 self.assertEqual('sessionmaster:2222', cluster_resolver.master())
139 cluster_resolver = TFConfigClusterResolver()
[all …]
|
| D | __init__.py | 21 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import ClusterResolver
22 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import SimpleClusterResolver
23 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import UnionClusterResolver
24 from tensorflow.python.distribute.cluster_resolver.gce_cluster_resolver import GCEClusterResolver
25 from tensorflow.python.distribute.cluster_resolver.kubernetes_cluster_resolver import KubernetesClu…
26 from tensorflow.python.distribute.cluster_resolver.slurm_cluster_resolver import SlurmClusterResolv…
27 from tensorflow.python.distribute.cluster_resolver.tfconfig_cluster_resolver import TFConfigCluster…
28 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| D | kubernetes_cluster_resolver_test.py | 21 from tensorflow.python.distribute.cluster_resolver import KubernetesClusterResolver
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 …]
|
| /external/tensorflow/tensorflow/contrib/cluster_resolver/python/training/ |
| D | __init__.py | 25 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import ClusterResolver
26 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import SimpleClusterResolver
27 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import UnionClusterResolver
28 from tensorflow.python.distribute.cluster_resolver.gce_cluster_resolver import GCEClusterResolver
29 from tensorflow.python.distribute.cluster_resolver.kubernetes_cluster_resolver import KubernetesClu…
30 from tensorflow.python.distribute.cluster_resolver.slurm_cluster_resolver import SlurmClusterResolv…
31 from tensorflow.python.distribute.cluster_resolver.tfconfig_cluster_resolver import TFConfigCluster…
32 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| D | cluster_resolver.py | 26 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import ClusterResolver
27 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import SimpleClusterResolver
28 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import UnionClusterResolver
|
| /external/tensorflow/tensorflow/contrib/cluster_resolver/ |
| D | __init__.py | 23 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import ClusterResolver
24 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import SimpleClusterResolver
25 from tensorflow.python.distribute.cluster_resolver.cluster_resolver import UnionClusterResolver
26 from tensorflow.python.distribute.cluster_resolver.gce_cluster_resolver import GCEClusterResolver
27 from tensorflow.python.distribute.cluster_resolver.kubernetes_cluster_resolver import KubernetesClu…
28 from tensorflow.python.distribute.cluster_resolver.slurm_cluster_resolver import SlurmClusterResolv…
29 from tensorflow.python.distribute.cluster_resolver.tfconfig_cluster_resolver import TFConfigCluster…
30 from tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver import TPUClusterResolver
|
| D | cluster_resolver_initialization_test.py | 21 from tensorflow.contrib.cluster_resolver import SimpleClusterResolver
22 from tensorflow.contrib.cluster_resolver.python.training import cluster_resolver
23 from tensorflow.contrib.cluster_resolver.python.training import UnionClusterResolver
35 cluster_resolver.SimpleClusterResolver(base_cluster_spec)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_strategy_util.py | 23 from tensorflow.python.distribute.cluster_resolver import TPUClusterResolver
37 def get_first_tpu_host_device(cluster_resolver): argument
50 if cluster_resolver.get_master() in ("", "local"):
52 job_name = cluster_resolver.get_job_name() or "tpu_worker"
57 def initialize_tpu_system(cluster_resolver=None): argument
66 if cluster_resolver is None:
67 cluster_resolver = TPUClusterResolver("")
91 with ops.device(get_first_tpu_host_device(cluster_resolver)):
96 master = cluster_resolver.master()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy.py | 36 from tensorflow.python.distribute.cluster_resolver import SimpleClusterResolver
37 from tensorflow.python.distribute.cluster_resolver import TFConfigClusterResolver
89 cluster_resolver=TFConfigClusterResolver()): argument
96 self._initialize_strategy(cluster_resolver)
100 def _initialize_strategy(self, cluster_resolver): argument
101 if cluster_resolver.cluster_spec().as_dict():
102 self._initialize_multi_worker(cluster_resolver)
104 self._initialize_local(cluster_resolver)
106 def _initialize_local(self, cluster_resolver): argument
113 if isinstance(cluster_resolver, TFConfigClusterResolver):
[all …]
|
| D | parameter_server_strategy.py | 32 from tensorflow.python.distribute.cluster_resolver import SimpleClusterResolver
33 from tensorflow.python.distribute.cluster_resolver import TFConfigClusterResolver
101 cluster_resolver=TFConfigClusterResolver()): argument
103 self._initialize_strategy(cluster_resolver)
109 def _initialize_strategy(self, cluster_resolver): argument
110 if cluster_resolver.cluster_spec().as_dict():
111 self._initialize_multi_worker(cluster_resolver)
113 self._initialize_local(cluster_resolver)
115 def _initialize_multi_worker(self, cluster_resolver): argument
132 if isinstance(cluster_resolver, TFConfigClusterResolver):
[all …]
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | collective_all_reduce_strategy.py | 24 from tensorflow.python.distribute.cluster_resolver import SimpleClusterResolver
25 from tensorflow.python.distribute.cluster_resolver import TFConfigClusterResolver
77 cluster_resolver = SimpleClusterResolver(
86 cluster_resolver=cluster_resolver)
|
| D | parameter_server_strategy.py | 24 from tensorflow.python.distribute.cluster_resolver import SimpleClusterResolver
25 from tensorflow.python.distribute.cluster_resolver import TFConfigClusterResolver
157 cluster_resolver = SimpleClusterResolver(
163 container_strategy, cluster_resolver=cluster_resolver)
|
| /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.-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…
|
| 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…
|
| /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.-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…
|
| 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.-t-p-u-cluster-resolver.pbtxt | 1 path: "tensorflow.distribute.cluster_resolver.TPUClusterResolver"
3 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.tpu_cluster_resolver.TPUClust…
4 …is_instance: "<class \'tensorflow.python.distribute.cluster_resolver.cluster_resolver.ClusterResol…
|