Exemplos de gpu_device_name em Python

Exemplo n.º 1

Arquivo: datasets_test.py Projeto: imdone/tensorflow

  def testTensorsExplicitPrefetchToDevice(self):
    ds = Dataset.from_tensor_slices([0., 1.])
    ds = ds.apply(prefetching_ops.prefetch_to_device(test.gpu_device_name()))

    with self.assertRaisesRegexp(TypeError, 'prefetch_to_device'):
      datasets.Iterator(ds)

    for i, x in enumerate(ds):
      with ops.device(test.gpu_device_name()):
        x = math_ops.add(x, x)
        self.assertEqual(float(i) + float(i), x.numpy())

Exemplo n.º 2

Arquivo: datasets_test.py Projeto: syed-ahmed/tensorflow

  def testTensorsExplicitPrefetchToDevice(self):
    ds = Dataset.from_tensor_slices([0., 1.])
    ds = ds.apply(prefetching_ops.prefetch_to_device(test.gpu_device_name()))

    with self.assertRaisesRegexp(TypeError, 'prefetch_to_device'):
      datasets.Iterator(ds)

    for i, x in enumerate(ds):
      with ops.device(test.gpu_device_name()):
        x = math_ops.add(x, x)
        self.assertEqual(float(i) + float(i), x.numpy())

Exemplo n.º 3

Arquivo: datasets_test.py Projeto: 568xiaoma/WeChatProgram

 def testTensorsPlacedOnDevice(self):
   ds = Dataset.from_tensors([0., 1.])
   with ops.device(test.gpu_device_name()):
     x = datasets.Iterator(ds).next()
     x = math_ops.add(x, x)
   self.assertAllEqual([0., 2.], x.numpy())

Exemplo n.º 4

Arquivo: datasets_test.py Projeto: zyy2020/tensorflow

 def testGpuDefinedDataset(self):
     with ops.device(test.gpu_device_name()):
         ds = Dataset.from_tensors([0., 1.])
         for x in ds:
             y = math_ops.add(x, x)
     self.assertAllEqual([0., 2.], y.numpy())

Exemplo n.º 5

Arquivo: datasets_test.py Projeto: aeverall/tensorflow

 def testGpuDefinedDataset(self):
   with ops.device(test.gpu_device_name()):
     ds = Dataset.from_tensors([0., 1.])
     for x in ds:
       y = math_ops.add(x, x)
   self.assertAllEqual([0., 2.], y.numpy())

Exemplo n.º 6

 def testGpuTensor(self):
     ds = dataset_ops.Dataset.from_tensors([0., 1.])
     with ops.device(test.gpu_device_name()):
         for x in ds:
             y = math_ops.add(x, x)
     self.assertAllEqual([0., 2.], y.numpy())