Searched refs:num_ps_tasks (Results 1 – 3 of 3) sorted by relevance
41 def __init__(self, num_ps_tasks, seed=0): argument43 self._num_tasks = num_ps_tasks
239 def parameter_server_scope(is_chief, ps_job_name, num_ps_tasks): argument275 "/job:%s/task:%s" % (ps_job_name, ps_index[0] % num_ps_tasks)):
128 is_chief=True, ps_job_name=JOB_NAME, num_ps_tasks=3):