Searched defs:ps_tasks (Results 1 – 1 of 1) sorted by relevance
78 def __init__(self, ps_tasks, ps_device, worker_device, merge_devices, ps_ops, argument137 def replica_device_setter(ps_tasks=0, argument