Method testOutputOnHostMemory

tensorflow/python/eager/ops_test.py:274–278

(self)

  @test_util.run_gpu_only
  def testOutputOnHostMemory(self):
    # The Shape op kernel on GPU places the output in host memory.
    value = constant_op.constant([1.]).gpu()
    shape = array_ops.shape(value)
    self.assertEqual([1], shape.numpy())

  @test_util.run_gpu_only
  def testSilentCopy(self):