Searched defs:ps_tasks (Results 1 – 1 of 1) sorted by relevance
75 def __init__(self, ps_tasks, ps_device, worker_device, merge_devices, ps_ops, argument134 def replica_device_setter(ps_tasks=0, ps_device="/job:ps", argument