Method take

tensorflow/python/ops/work_queue.py:540–545

Take work from the local workqueue.

(self)

              collections=[ops.GraphKeys.LOCAL_VARIABLES])

  def take(self):
    """Take work from the local workqueue."""
    with ops.name_scope(self._name):
      with ops.device(self._local_device):
        with ops.device('/cpu:0'):
          return self._restore_work_queue.dequeue()

  def hook(self):
    """Get a Hook that controls the restore of the works."""