tfqueue.py 文件源码

def setup(self, graph):
        self._placeholders = graph.get_collection(TArtGraphKeys.PLACEHOLDERS)
        placeholders_dtypes = [x.dtype for x in self._placeholders]
        self._input_queue = tf.FIFOQueue(self._env.flags.input_queue_size, placeholders_dtypes, name=self._name)
        self._input_queue_cond = tf.placeholder_with_default(True, shape=[], name=self._name + '_cond')

        self.enqueue_op = self._input_queue.enqueue(self._placeholders)
        self.dequeue_op = self._input_queue.dequeue()
        self.close_op = self._input_queue.close(cancel_pending_enqueues=True)
        self.qsize_op = self._input_queue.size()

        for a, b in zip(self._placeholders, self.dequeue_op):
            as_tftensor(b).set_shape(as_tftensor(a).get_shape())

        self.edit_graph(graph)