Method test_simple

tensorflow/python/ops/prefetch_test.py:39–55

(self)

# pylint: disable=missing-docstring
class PrefetchTest(test.TestCase):
  def test_simple(self):
    capacity = 2
    value = 42.0
    with ops.Graph().as_default() as graph:
      with ops.device('/cpu:0'):
        x = array_ops.constant(value, dtype=dtypes.float32, shape=[])
      with ops.device(test.gpu_device_name()):
        y = prefetch.staged(x, capacity=capacity, num_threads=2, timeout_millis=1000)

    graph.finalize()

    with self.test_session(use_gpu=True, graph=graph) as sess:
      coord = coordinator.Coordinator()
      prefetch.make_prefetch_hook().after_create_session(sess, coord)
      for _ in xrange(capacity * 3):
        self.assertAllClose(value, sess.run(y), rtol=1e-6)
      coord.request_stop()

  def test_string(self):
    capacity = 3