Home
last modified time | relevance | path

Searched refs:_cluster_spec (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dserver_lib.py286 self._cluster_spec = {}
295 self._cluster_spec[job_name] = job_tasks
299 self._cluster_spec = {}
301 self._cluster_spec[job_def.name] = {
307 self._cluster_spec = {}
309 self._cluster_spec[job_def.name] = {
318 return bool(self._cluster_spec)
324 return self._cluster_spec == other
327 return self._cluster_spec != other
374 return list(self._cluster_spec.keys())
[all …]
Ddevice_setter_test.py32 _cluster_spec = server_lib.ClusterSpec({ variable in 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/
Dmulti_process_runner.py152 self._cluster_spec = cluster_spec
187 if 'chief' in self._cluster_spec:
192 for _ in range(len(nest.flatten(self._cluster_spec)) - 1):
313 for task_type, addresses in self._cluster_spec.items():
317 args=(self._proc_func, task_type, task_id, self._cluster_spec) +
351 self._cluster_spec = updated_cluster_spec or self._cluster_spec
355 args=(proc_func, task_type, task_id, self._cluster_spec) + (args or ()),
Destimator_training.py95 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
Ddistribute_coordinator.py138 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.EVALUATOR
[all …]
Dparameter_server_strategy.py251 self._cluster_spec = cluster_spec
301 self._cluster_spec = None
332 if self._cluster_spec:
334 self._cluster_spec, self._task_type, self._task_id)
336 self._cluster_spec, self._task_type)
353 if self._cluster_spec:
355 self._cluster_spec, self._task_type, self._task_id)
357 self._cluster_spec, self._task_type)
458 if not self._cluster_spec:
584 if not self._cluster_spec:
Dparameter_server_strategy_test.py112 cluster_spec=self._cluster_spec,
370 if d.extended._cluster_spec:
371 num_workers = len(d.extended._cluster_spec.as_dict().get(WORKER))
372 if 'chief' in d.extended._cluster_spec.as_dict():
436 assert hasattr(d.extended, '_cluster_spec') and d.extended._cluster_spec
437 num_workers = len(d.extended._cluster_spec.as_dict().get(WORKER))
438 if CHIEF in d.extended._cluster_spec.as_dict():
493 d.extended._cluster_spec, task_type, task_id)):
569 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
571 cls._default_target = 'grpc://' + cls._cluster_spec[WORKER][0]
[all …]
Dcollective_all_reduce_strategy_test.py106 cluster_spec=self._cluster_spec,
399 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
405 cluster_spec=self._cluster_spec,
409 num_workers = len(self._cluster_spec.get('chief', []) +
410 self._cluster_spec.get('worker', []))
418 self._cluster_spec, required_gpus)
425 self._cluster_spec,
432 self._test_complex_model, self._cluster_spec, num_gpus=required_gpus)
443 self._cluster_spec,
531 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
[all …]
Dcross_device_ops_test.py444 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
494 cluster_spec=self._cluster_spec,
500 "grpc://" + self._cluster_spec[task_type][task_id])
506 "grpc://" + self._cluster_spec[task_type][task_id])
549 num_workers = len(self._cluster_spec.get("chief", [])) + len(
550 self._cluster_spec.get("worker", []))
666 num_workers = len(self._cluster_spec.get("chief", [])) + len(
667 self._cluster_spec.get("worker", []))
713 self._cluster_spec,
723 self._cluster_spec, required_gpus,
Dcollective_all_reduce_strategy.py226 self._cluster_spec = None
253 self._cluster_spec = cluster_spec
397 if self._cluster_spec is None:
401 self._cluster_spec, self._task_type, self._task_id)
489 if not self._cluster_spec:
498 multi_worker_util.collective_leader(self._cluster_spec, self._task_type,
Ddistribute_coordinator_test.py173 cls._cluster_spec = {
436 cluster_spec=self._cluster_spec)
444 cluster_spec=self._cluster_spec)
455 cluster_spec=self._cluster_spec)
465 cluster_spec=self._cluster_spec)
489 cluster_spec=self._cluster_spec)
507 cluster_spec=self._cluster_spec)
538 cluster_spec = copy.deepcopy(self._cluster_spec)
571 cluster_spec = copy.deepcopy(self._cluster_spec)
Dmirrored_strategy_test.py1242 cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
1244 cls._default_target = "grpc://" + cls._cluster_spec["chief"][0]
1255 strategy.configure(cluster_spec=self._cluster_spec)
1263 strategy.configure(cluster_spec=self._cluster_spec)
1269 cluster_spec["chief"] = self._cluster_spec["chief"]
1285 tf_config = {"cluster": self._cluster_spec}
1298 strategy.configure(cluster_spec=self._cluster_spec)
Dmulti_worker_test_base.py250 cls._cluster_spec = create_in_process_cluster(num_workers=num_workers,
252 cls._default_target = 'grpc://' + cls._cluster_spec['worker'][0]
Dinput_lib_test.py871 "cluster": self._cluster_spec,
881 target="grpc://" + self._cluster_spec[task_type][task_id]) as sess:
918 self._run_between_graph_clients(_worker_fn, self._cluster_spec, 0)
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver.py194 self._cluster_spec = cluster_spec
202 return self._cluster_spec