Python get_log_device_placement 예제들

예제 #1

파일: config_test.py 프로젝트: waterdrops/tensorflow

    def testLogDevicePlacement(self):
        self.assertFalse(context.get_log_device_placement())

        context.set_log_device_placement(True)
        self.assertEqual(context.get_log_device_placement(), True)
        self.assertEqual(context.get_log_device_placement(),
                         context.context().log_device_placement)

        context.set_log_device_placement(False)
        self.assertEqual(context.get_log_device_placement(), False)
        self.assertEqual(context.get_log_device_placement(),
                         context.context().log_device_placement)

        context.ensure_initialized()

        # Changing the device placement should not throw an exception
        context.set_log_device_placement(True)

예제 #2

파일: config_test.py 프로젝트: zuozi2810/tensorflow

    def testLogDevicePlacement(self):
        self.assertEqual(context.get_log_device_placement(), False)

        context.set_log_device_placement(True)
        self.assertEqual(context.get_log_device_placement(), True)
        self.assertEqual(context.get_log_device_placement(),
                         context.context().log_device_placement)

        context.set_log_device_placement(False)
        self.assertEqual(context.get_log_device_placement(), False)
        self.assertEqual(context.get_log_device_placement(),
                         context.context().log_device_placement)

        constant_op.constant(1)
        with self.assertRaises(RuntimeError):
            context.set_log_device_placement(True)
        with self.assertRaises(RuntimeError):
            context.set_log_device_placement(False)

예제 #3

파일: config_test.py 프로젝트: zhujinzhenlmu/tensorflow

    def testLogDevicePlacement(self):
        self.assertFalse(context.get_log_device_placement())

        context.set_log_device_placement(True)
        self.assertEqual(context.get_log_device_placement(), True)
        self.assertEqual(context.get_log_device_placement(),
                         context.context().log_device_placement)

        context.set_log_device_placement(False)
        self.assertEqual(context.get_log_device_placement(), False)
        self.assertEqual(context.get_log_device_placement(),
                         context.context().log_device_placement)

        context.ensure_initialized()

        with self.assertRaises(RuntimeError):
            context.set_log_device_placement(True)
        with self.assertRaises(RuntimeError):
            context.set_log_device_placement(False)

예제 #4

파일: config_test.py 프로젝트: kylin9872/tensorflow

  def testLogDevicePlacement(self):
    self.assertEqual(context.get_log_device_placement(), False)

    context.set_log_device_placement(True)
    self.assertEqual(context.get_log_device_placement(), True)
    self.assertEqual(
        context.get_log_device_placement(),
        context.context().log_device_placement)

    context.set_log_device_placement(False)
    self.assertEqual(context.get_log_device_placement(), False)
    self.assertEqual(
        context.get_log_device_placement(),
        context.context().log_device_placement)

    constant_op.constant(1)
    with self.assertRaises(RuntimeError):
      context.set_log_device_placement(True)
    with self.assertRaises(RuntimeError):
      context.set_log_device_placement(False)

예제 #5

파일: config_test.py 프로젝트: Harryi0/tinyML

    def testLogDevicePlacement(self):
        self.assertFalse(context.get_log_device_placement())

        context.set_log_device_placement(True)
        self.assertEqual(context.get_log_device_placement(), True)
        self.assertEqual(context.get_log_device_placement(),
                         context.context().log_device_placement)

        context.set_log_device_placement(False)
        self.assertEqual(context.get_log_device_placement(), False)
        self.assertEqual(context.get_log_device_placement(),
                         context.context().log_device_placement)

        context.ensure_initialized()

        with self.assertRaises(RuntimeError):
            context.set_log_device_placement(True)

        # If the setting the device placement is a no-op, do not throw a runtime
        # exception.
        context.set_log_device_placement(False)

예제 #6

파일: config_test.py 프로젝트: adit-chandra/tensorflow

  def testLogDevicePlacement(self):
    self.assertFalse(context.get_log_device_placement())

    context.set_log_device_placement(True)
    self.assertEqual(context.get_log_device_placement(), True)
    self.assertEqual(
        context.get_log_device_placement(),
        context.context().log_device_placement)

    context.set_log_device_placement(False)
    self.assertEqual(context.get_log_device_placement(), False)
    self.assertEqual(
        context.get_log_device_placement(),
        context.context().log_device_placement)

    context.ensure_initialized()

    with self.assertRaises(RuntimeError):
      context.set_log_device_placement(True)
    with self.assertRaises(RuntimeError):
      context.set_log_device_placement(False)

예제 #7

파일: config_test.py 프로젝트: aritratony/tensorflow

  def testLogDevicePlacement(self):
    self.assertFalse(context.get_log_device_placement())

    context.set_log_device_placement(True)
    self.assertEqual(context.get_log_device_placement(), True)
    self.assertEqual(
        context.get_log_device_placement(),
        context.context().log_device_placement)

    context.set_log_device_placement(False)
    self.assertEqual(context.get_log_device_placement(), False)
    self.assertEqual(
        context.get_log_device_placement(),
        context.context().log_device_placement)

    context.ensure_initialized()

    with self.assertRaises(RuntimeError):
      context.set_log_device_placement(True)

    # If the setting the device placement is a no-op, do not throw a runtime
    # exception.
    context.set_log_device_placement(False)