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