OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:ps_tasks
(Results
1 - 9
of
9
) sorted by null
/external/tensorflow/tensorflow/python/training/
device_setter.py
67
def __init__(self,
ps_tasks
, ps_device, worker_device, merge_devices, ps_ops,
72
ps_tasks
: Number of tasks in the `ps` job.
82
self._ps_tasks =
ps_tasks
106
# present and
ps_tasks
is non-zero. However, its task number will only be
126
def replica_device_setter(
ps_tasks
=0, ps_device="/job:ps",
137
If `cluster` is `None`, and `
ps_tasks
` is 0, the returned function is a no-op.
138
Otherwise, the value of `
ps_tasks
` is derived from `cluster`.
162
ps_tasks
: Number of tasks in the `ps` job. Ignored if `cluster` is
196
ps_tasks
= len(cluster_spec[ps_job_name])
198
if
ps_tasks
== 0
[
all
...]
localhost_cluster_performance_test.py
93
with ops.device(device_setter.replica_device_setter(
ps_tasks
=100)):
device_setter_test.py
87
device_setter.replica_device_setter(
ps_tasks
=1, ps_device="/cpu:0")):
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
gbdt_batch.py
[
all
...]
/external/tensorflow/tensorflow/contrib/opt/python/training/
elastic_average_optimizer_test.py
77
ps_tasks
=1)):
model_average_optimizer_test.py
77
ps_tasks
=1)):
/external/tensorflow/tensorflow/python/estimator/
estimator.py
[
all
...]
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
estimator.py
224
ps_tasks
=config.num_ps_replicas,
[
all
...]
/external/tensorflow/tensorflow/contrib/framework/python/ops/
variables_test.py
455
with ops.device(device_setter.replica_device_setter(
ps_tasks
=2)):
[
all
...]
Completed in 174 milliseconds