| /external/tensorflow/tensorflow/python/training/ |
| server_lib.py | 285 self._cluster_spec = {}
294 self._cluster_spec[job_name] = job_tasks
298 self._cluster_spec = {}
300 self._cluster_spec[job_def.name] = {
305 self._cluster_spec = {}
307 self._cluster_spec[job_def.name] = {
315 return bool(self._cluster_spec)
321 return self._cluster_spec == other
324 return self._cluster_spec != other
370 return list(self._cluster_spec.keys()
[all...] |
| device_setter_test.py | 32 _cluster_spec = server_lib.ClusterSpec({ variable in class:DeviceSetterTest
40 device_setter.replica_device_setter(cluster=self._cluster_spec)):
55 device_setter.replica_device_setter(cluster=self._cluster_spec)):
62 device_setter.replica_device_setter(cluster=self._cluster_spec)):
75 device_setter.replica_device_setter(cluster=self._cluster_spec)):
108 cluster=self._cluster_spec, merge_devices=False)):
122 device_setter.replica_device_setter(cluster=self._cluster_spec.as_dict(
137 cluster=self._cluster_spec.as_cluster_def())):
|
| /external/tensorflow/tensorflow/python/distribute/ |
| estimator_training.py | 95 config._cluster_spec = worker_context.cluster_spec
102 if config._cluster_spec:
106 config._num_ps_replicas = _count_ps(config._cluster_spec)
108 config._cluster_spec, chief_task_type=CHIEF)
110 config._cluster_spec,
116 config._cluster_spec = server_lib.ClusterSpec({})
155 config._cluster_spec = config._experimental_distribute.remote_cluster
175 config._cluster_spec = cluster_spec
|
| distribute_coordinator.py | 138 self._cluster_spec = cluster_spec
149 if self._cluster_spec:
151 self._cluster_spec, self.task_type, self.task_id)
172 if not self._cluster_spec:
178 if _TaskType.CHIEF in self._cluster_spec.jobs:
182 assert _TaskType.WORKER in self._cluster_spec.jobs
192 return prefix + self._cluster_spec.job_tasks(task_type)[task_id or 0]
196 if (not self._cluster_spec or
202 if (_TaskType.CHIEF not in self._cluster_spec.jobs and
278 return bool(self._cluster_spec) and self._task_type != _TaskType.EVALUATO
[all...] |
| collective_all_reduce_strategy.py | 133 self._cluster_spec = None
203 self._cluster_spec = cluster_spec
340 if self._cluster_spec is None:
344 self._cluster_spec, self._task_type, self._task_id)
405 if not self._cluster_spec:
413 if "chief" in self._cluster_spec.jobs:
417 if "worker" not in self._cluster_spec.jobs:
|
| parameter_server_strategy.py | 194 self._cluster_spec = cluster_spec
244 self._cluster_spec = None
264 if self._cluster_spec:
266 self._cluster_spec, self._task_type, self._task_id)
268 self._cluster_spec, self._task_type)
369 if not self._cluster_spec:
495 if not self._cluster_spec:
|
| distribute_coordinator_test.py | 170 cls._cluster_spec = {
417 cluster_spec=self._cluster_spec)
425 cluster_spec=self._cluster_spec)
436 cluster_spec=self._cluster_spec)
446 cluster_spec=self._cluster_spec)
470 cluster_spec=self._cluster_spec)
488 cluster_spec=self._cluster_spec)
519 cluster_spec = copy.deepcopy(self._cluster_spec)
552 cluster_spec = copy.deepcopy(self._cluster_spec)
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| run_config.py | 144 self._cluster_spec = server_lib.ClusterSpec(config.get('cluster', {}))
146 _get_master(self._cluster_spec, self._task_type,
148 self._num_ps_replicas = _count_ps(self._cluster_spec) or 0
149 self._num_worker_replicas = _count_worker(self._cluster_spec) or 0
170 return self._cluster_spec
363 if '_cluster_spec' in ordered_state:
364 ordered_state['_cluster_spec'] = collections.OrderedDict(
365 sorted(ordered_state['_cluster_spec'].as_dict().items(),
|
| linear_test.py | 499 config._cluster_spec = server_lib.ClusterSpec({})
[all...] |
| dnn_linear_combined_test.py | 506 config._cluster_spec = server_lib.ClusterSpec({})
[all...] |
| dnn_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| parameter_server_strategy_test.py | 140 cluster_spec=self._cluster_spec,
409 if d.extended._cluster_spec:
410 num_workers = len(d.extended._cluster_spec.as_dict().get(WORKER))
411 if 'chief' in d.extended._cluster_spec.as_dict():
485 assert hasattr(d.extended, '_cluster_spec') and d.extended._cluster_spec
486 num_workers = len(d.extended._cluster_spec.as_dict().get(WORKER))
487 if CHIEF in d.extended._cluster_spec.as_dict():
545 d.extended._cluster_spec, task_type, task_id)):
616 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster
[all...] |
| collective_all_reduce_strategy_test.py | 129 cluster_spec=self._cluster_spec,
356 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
363 cluster_spec=self._cluster_spec,
368 num_workers = len(self._cluster_spec.get('chief', []) +
369 self._cluster_spec.get('worker', []))
382 self._cluster_spec,
397 self._cluster_spec,
412 self._cluster_spec,
509 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
516 self._cluster_spec, num_gpus
[all...] |
| cross_device_ops_test.py | 444 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
497 cluster_spec=self._cluster_spec,
503 "grpc://" + self._cluster_spec[task_type][task_id])
508 "grpc://" + self._cluster_spec[task_type][task_id])
544 num_workers = len(self._cluster_spec.get("chief", [])) + len(
545 self._cluster_spec.get("worker", []))
650 self._cluster_spec,
|
| estimator_training_test.py | 80 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
283 cluster_spec = copy.deepcopy(self._cluster_spec)
313 cluster_spec = copy.deepcopy(self._cluster_spec)
342 cluster = copy.deepcopy(self._cluster_spec)
|
| keras_multi_worker_test.py | 304 cls._cluster_spec = test_base.create_in_process_cluster(
328 _run_standalone_client(self, strategy, self._cluster_spec)
|
| multi_worker_test_base.py | 205 cls._cluster_spec = create_in_process_cluster(num_workers=2, num_ps=0)
206 cls._default_target = 'grpc://' + cls._cluster_spec['worker'][0]
|
| mirrored_strategy_multigpu_test.py | [all...] |
| /external/tensorflow/tensorflow/python/tpu/ |
| tpu_config.py | 239 self._cluster_spec = cluster.cluster_spec()
249 if self._cluster_spec:
251 self._cluster_spec.as_cluster_def())
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| cluster_resolver.py | 200 self._cluster_spec = cluster_spec
208 return self._cluster_spec
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| experiment_test.py | 194 def _cluster_spec(self): member in class:ExperimentTest
284 'cluster': self._cluster_spec(),
338 tf_config = {'cluster': self._cluster_spec()}
353 'cluster': self._cluster_spec(),
834 'cluster': self._cluster_spec(),
[all...] |
