Method testDevice

tensorflow/compiler/xla/python/xla_client_test.py:627–632

(self)

    np.testing.assert_equal(arg0, arg0_buffer.to_py())

  def testDevice(self):
    x = np.arange(8)
    for device in xla_client.get_local_backend().local_devices():
      buf = xla_client.Buffer.from_pyval(x, device=device)
      self.assertEqual(buf.device(), device)
      np.testing.assert_equal(x, buf.to_py())


class SingleOpTest(ComputationTest):