Method testCopyBetweenDevices

tensorflow/python/eager/core_test.py:519–528

(self)

  @test_util.run_gpu_only
  def testCopyBetweenDevices(self):
    x = constant_op.constant([[1., 2.], [3., 4.]])
    x = x.cpu()
    x = x.gpu()
    x = x.gpu()
    x = x.cpu()

    # Invalid device
    with self.assertRaises(RuntimeError):
      x.gpu(context.context().num_gpus() + 1)

  @test_util.run_gpu_only
  def testCopyBetweenDevicesAsync(self):