def connect_done_queue(cluster_spec, task_index):
done_ops = []
for i in range(cluster_spec.num_tasks("ps")):
with tf.device("/job:ps/task:%d" % i):
queue = tf.FIFOQueue(cluster_spec.num_tasks('worker'), tf.int32,
shared_name='done_queue' + str(i))
done_ops.append(queue.enqueue(task_index))
return done_ops
def create_done_queue(cluster_spec, task_index):
with tf.device("/job:ps/task:%d" % (task_index)):
queue = tf.FIFOQueue(cluster_spec.num_tasks("worker"),
tf.int32,
shared_name="done_queue" + str(task_index))
return queue