Searched refs:ps_job_name (Results 1 – 3 of 3) sorted by relevance
200 ps_job_name = pydev.DeviceSpec.from_string(ps_device).job201 if ps_job_name not in cluster_spec or cluster_spec[ps_job_name] is None:203 ps_tasks = len(cluster_spec[ps_job_name])
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):