Method testInt32GPU

tensorflow/python/eager/core_test.py:415–419

(self)

  @test_util.run_gpu_only
  def testInt32GPU(self):
    with ops.device('gpu:0'):
      xent = nn_ops.sparse_softmax_cross_entropy_with_logits(
          logits=[[0.0, 0.0]], labels=[0])
    self.assertAllClose(xent, [0.69314718])

  def _runInThread(self, target, args):
    t = threading.Thread(target=target, args=args)