HomeSort by relevance Sort by last modified time
    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